Skip to content

Commit 1fd2a7e

Browse files
committed
new string format for schedule for legibility (apache#8)
1 parent 56c57e5 commit 1fd2a7e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vta/python/vta/top/vta_conv2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -379,7 +379,7 @@ def _traverse(op):
379379
pad_data = None
380380
wrkld = _get_workload(data, pad_data, kernel, output)
381381
if planStr:
382-
matchObj = re.match( r'b(\d+)oc(\d+)ic(\d+)h(\d+)w(\d+)oc_t(\d+)h_t(\d+)', sched_str)
382+
matchObj = re.match( r'b(\d+)_oc(\d+)_ic(\d+)_h(\d+)_w(\d+)_oct(\d+)_ht(\d+)', sched_str)
383383
b_factor = int(matchObj.group(1))
384384
oc_factor = int(matchObj.group(2))
385385
ic_factor = int(matchObj.group(3))

0 commit comments

Comments
 (0)