From 709718ff108d125348bb10ef2958b399e988cf50 Mon Sep 17 00:00:00 2001 From: root Date: Sun, 28 Jul 2024 12:54:32 -0700 Subject: [PATCH] fix max size of tile calc --- sam/onyx/generate_matrices.py | 2 +- sam/onyx/parse_dot.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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]