add utils

KULeuven-MICAS · Oct 29, 2024 · c282026 · c282026
1 parent a1b9522
commit c282026
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/target/sim/sw/device/apps/snax/snax-gemmx-conv/data/datagen.py b/target/sim/sw/device/apps/snax/snax-gemmx-conv/data/datagen.py
@@ -13,11 +13,20 @@
 import sys
 import os
 
+import subprocess
+
 # Add data utility path
-sys.path.append(os.path.join(os.path.dirname(__file__), "../../../../../../util/sim/"))
+sys.path.append(os.path.join(os.path.dirname(__file__),
+                "../../../../../../../../util/sim/"))
 from data_utils import format_scalar_definition, format_vector_definition  # noqa E402
 
 # Add golden model path
+bender_command = subprocess.run(['bender', 'path', 'snitch_cluster'],
+                                capture_output=True, text=True)
+snax_utils_path = bender_command.stdout.strip()
+
+sys.path.append(snax_utils_path + "/util/sim/")
+
 from snax_utils import (  # noqa E402
     conv2d,
     im2col,

diff --git a/target/sim/sw/device/apps/snax/snax-gemmx-matmul/data/datagen.py b/target/sim/sw/device/apps/snax/snax-gemmx-matmul/data/datagen.py
@@ -13,11 +13,19 @@
 import sys
 import os
 
+import subprocess
+
 # Add data utility path
-sys.path.append(os.path.join(os.path.dirname(__file__), "../../../../../../util/sim/"))
+sys.path.append(os.path.join(os.path.dirname(__file__),
+                "../../../../../../../../util/sim/"))
 from data_utils import format_scalar_definition, format_vector_definition  # noqa E402
 
 # Add golden model path
+bender_command = subprocess.run(['bender', 'path', 'snitch_cluster'],
+                                capture_output=True, text=True)
+snax_utils_path = bender_command.stdout.strip()
+
+sys.path.append(snax_utils_path + "/util/sim/")
 from snax_utils import (  # noqa E402
     conv2d,
     im2col,

diff --git a/target/sim/sw/sim_elf.yaml b/target/sim/sw/sim_elf.yaml
@@ -4,4 +4,5 @@
 
 runs:
   - elf: host/apps/offload/build/offload-snax-test-integration.elf
-  - elf: host/apps/offload/build/offload-snax-gemmx.elf
+  - elf: host/apps/offload/build/offload-snax-gemmx-matmul.elf
+  - elf: host/apps/offload/build/offload-snax-gemmx-conv.elf