Moved test_utils.py to utils/test.py (#1270)

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Xilinx · Apr 17, 2024 · 15fe206 · 15fe206
1 parent 8e358fe
commit 15fe206
Show file tree

Hide file tree

Showing 9 changed files with 968 additions and 73 deletions.
diff --git a/programming_examples/basic/eltwise_exp/test.py b/programming_examples/basic/eltwise_exp/test.py
@@ -9,8 +9,7 @@
 import sys
 import time
 
-sys.path.append("../../programming_examples/utils")
-import test_utils
+import aie.utils.test as test_utils
 
 # ------------------------------------------------------
 # Configure this to match your design's buffer size

diff --git a/programming_examples/basic/eltwise_mul/test.py b/programming_examples/basic/eltwise_mul/test.py
@@ -11,8 +11,7 @@
 import sys
 import time
 
-sys.path.append("../../python")
-import test_utils
+import aie.utils.test as test_utils
 
 # ------------------------------------------------------
 # Configure this to match your design's buffer size

diff --git a/programming_examples/ml/eltwise_add/test.py b/programming_examples/ml/eltwise_add/test.py
@@ -12,8 +12,7 @@
 import sys
 import time
 
-sys.path.append("../../programming_examples/utils")
-import test_utils
+import aie.utils.test as test_utils
 
 # ------------------------------------------------------
 # Configure this to match your design's buffer size

diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
@@ -26,6 +26,15 @@ declare_mlir_python_sources(AIEPythonSources.Util
     util.py
 )
 
+declare_mlir_python_sources(AIEPythonSources.Utils
+  ADD_TO_PARENT AIEPythonSources
+  SOURCES
+    utils/test.py
+    utils/xrt.py
+    utils/ml.py
+    utils/trace.py
+)
+
 declare_mlir_python_sources(AIEPythonSources.Extras
   ADD_TO_PARENT AIEPythonSources
   SOURCES_GLOB

diff --git a/python/test_utils.py b/python/test_utils.py