diff --git a/sam/onyx/generate_matrices.py b/sam/onyx/generate_matrices.py index bb9d4dcf..7d2fb118 100644 --- 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 index ecf3f39f..1fd818ec 100644 --- 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]