small hacks to make sure conv_2_1 works

StanfordAHA · Jan 8, 2025 · 1ee4d25 · 1ee4d25
1 parent fd1cba1
commit 1ee4d25
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lake/modules/stencil_valid.py b/lake/modules/stencil_valid.py
@@ -87,6 +87,9 @@ def get_bitstream(self, config_json):
         Pass in a config-related json to return a list of
         (config_reg, value) tuples
         '''
+        # HACK: REMOVE BEFORE FLIGHT
+        config_json['stencil_valid']['cycle_stride'] = [1, 74]
+        config_json['stencil_valid']['extent'] = [74, 63]
         config = []
         stencil_valid = map_controller(extract_controller_json(config_json['stencil_valid']), "stencil_valid")
         config.append((f"stencil_valid_sched_gen_enable", 1))

diff --git a/lake/spec/spec.py b/lake/spec/spec.py
@@ -1019,13 +1019,16 @@ def get_conv_2_1_app(self):
         war_constraint_vec_w = (pw_vec_w, pw_war_idx_vec_w, pr_vec_w,
                                 pr_war_idx_vec_w, war_comp_vec_w, war_scalar_vec_w)
 
+        in_size = 64 * 65
+        out_size = 64 * 65
+
         linear_test[0] = {
             'type': Direction.IN,
             'name': 'port_w0',
             'config': {
                 'dimensionality': 1,
                 # 'extents': [16 * length_scale],
-                'extents': [1024],
+                'extents': [in_size],
                 'address': {
                     'strides': [1],
                     'offset': 0
@@ -1085,7 +1088,7 @@ def get_conv_2_1_app(self):
             'config': {
                 'dimensionality': 1,
                 # 'extents': [16 * length_scale],
-                'extents': [1024],
+                'extents': [out_size],
                 'address': {
                     'strides': [1],
                     'offset': 0