Merge pull request #141 from weiya711/glb_batch_size

fix max size of tile calc
weiya711 · Jul 29, 2024 · e7a5129 · e7a5129
2 parents 516b686 + 709718f
commit e7a5129
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sam/onyx/generate_matrices.py b/sam/onyx/generate_matrices.py
@@ -494,7 +494,7 @@ def create_matrix_from_point_list(name, pt_list, shape, use_fp=False, base=16) -
     return mg
 
 
-def convert_aha_glb_output_file(glbfile, output_dir, tiles, batches, glb_mem_stride=500):
+def convert_aha_glb_output_file(glbfile, output_dir, tiles, batches, glb_mem_stride=1024):
 
     glbfile_s = os.path.basename(glbfile).rstrip(".txt")
 

diff --git a/sam/onyx/parse_dot.py b/sam/onyx/parse_dot.py
@@ -19,7 +19,7 @@ def __init__(self, *args: object) -> None:
 class SAMDotGraph():
 
     def __init__(self, filename=None, local_mems=True, use_fork=False,
-                 use_fa=False, unroll=1, collat_dir=None, opal_workaround=False, mem_block_size=1000) -> None:
+                 use_fa=False, unroll=1, collat_dir=None, opal_workaround=False, mem_block_size=2048) -> None:
         assert filename is not None, "filename is None"
         self.graphs = pydot.graph_from_dot_file(filename)
         self.graph = self.graphs[0]