Merge pull request #965 from StanfordAHA/sparse_tiling

tiling sparse matrices fixes
StanfordAHA · May 31, 2023 · 246cc43 · 246cc43
2 parents 583c025 + 81dedf0
commit 246cc43
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/SPARSE_TESTS/copy_formatted.py b/SPARSE_TESTS/copy_formatted.py
@@ -3,8 +3,13 @@
 import subprocess
 import os
 
-b_tensors = glob.glob("../matmul_ijk_ch7-6-b1/formatted/tensor_B*")
-c_tensors = glob.glob("../matmul_ijk_ch7-6-b1/formatted/tensor_C*")
+test = "relat5_tile20" 
+
+b_tensors = glob.glob(f"../matmul_ijk_{test}/formatted/tensor_B*")
+c_tensors = glob.glob(f"../matmul_ijk_{test}/formatted/tensor_C*")
+
+if not os.path.exists("./MAT_TMP_DIR"):
+    os.mkdir("./MAT_TMP_DIR")
 
 os.system("rm -rf ./matmul*")
 os.system("rm -rf ./MAT_TMP_DIR/tile*")
@@ -34,11 +39,11 @@
 
             shutil.copy(f"{b}/B_shape.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_B_mode_shape")
 
-            shutil.copy(f"{c}/C0_crd.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_1_crd")
-            shutil.copy(f"{c}/C0_seg.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_1_seg")
+            shutil.copy(f"{c}/C0_crd.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_0_crd")
+            shutil.copy(f"{c}/C0_seg.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_0_seg")
 
-            shutil.copy(f"{c}/C1_crd.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_0_crd")
-            shutil.copy(f"{c}/C1_seg.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_0_seg")
+            shutil.copy(f"{c}/C1_crd.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_1_crd")
+            shutil.copy(f"{c}/C1_seg.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_1_seg")
 
             shutil.copy(f"{c}/C_vals.txt", f"./MAT_TMP_DIR/{tile_str}/tensor_C_mode_vals")
 

diff --git a/tests/test_memory_core/build_tb.py b/tests/test_memory_core/build_tb.py
@@ -1523,7 +1523,7 @@ def prepare_glb_collateral(glb_dir=None, bitstream=None, matrices_in=None, desig
                         "dimensionality": 1,
                         # "extent": [num_lines],
                         # HAX for testing against GLB
-                        "extent": [num_lines * 10 + 500],
+                        "extent": [2048],
                         "write_data_starting_addr": [0],
                         "write_data_stride": [1]
                     },