new string format for schedule for legibility (apache#8)

tmoreau89 · Jan 2, 2019 · 1fd2a7e · 1fd2a7e
1 parent 56c57e5
commit 1fd2a7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vta/python/vta/top/vta_conv2d.py b/vta/python/vta/top/vta_conv2d.py
@@ -379,7 +379,7 @@ def _traverse(op):
  pad_data = None
  wrkld = _get_workload(data, pad_data, kernel, output)
  if planStr:
- matchObj = re.match( r'b(\d+)oc(\d+)ic(\d+)h(\d+)w(\d+)oc_t(\d+)h_t(\d+)', sched_str)
+ matchObj = re.match( r'b(\d+)_oc(\d+)_ic(\d+)_h(\d+)_w(\d+)_oct(\d+)_ht(\d+)', sched_str)
  b_factor = int(matchObj.group(1))
  oc_factor = int(matchObj.group(2))
  ic_factor = int(matchObj.group(3))