ready for resnet benchmark

compiler/ir/autoflow/scheduler.py

@@ -96,6 +96,10 @@ def generate_one_list(n: int, i: int):
                     if schedule_bound % template_bound != 0:
                         pass
                         print('imperfect factorization, no support yet')
+                        padded_schedule = schedule.pad_dim(N, schedule_bound + (schedule_bound % template_bound))
+                        tiled_schedule = padded_schedule.tile_dim(N, template_bound)
+                        # try again with padded schedule, but no increased dim
+                        yield from scheduler_backtrack(template, tiled_schedule, pure_output_stationary, dim + 1)
                     else:
                         pass
                         print('match, will apply tiling')
@@ -117,5 +121,7 @@ def scheduler(template: Template, schedule: Schedule, schedule_idx: int = 0, pur
 
     schedules = list(schedules)
 
+    breakpoint()
+
     # match at schedule idx
     return schedules[schedule_idx]

kernels/conv/genbenchmark.py

@@ -92,7 +92,7 @@ def generate_resnet_benchmark():
     ]
 
 
-    for i, spec in enumerate(specs[0:1]):
+    for i, spec in enumerate(specs):
 
         module = generate_conv_ir(spec, generate_constants = False)