Skip to content

Commit

Permalink
Fix returned error for winograd-conv. Update returned error for direc…
Browse files Browse the repository at this point in the history
…t-conv similarly.
  • Loading branch information
psyhtest committed Jul 20, 2018
1 parent bfd8554 commit 2999c74
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions script/process-nntest-conv/process.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,10 +31,10 @@ def ck_preprocess(i):
stride = str(env.get('CK_CONV_STRIDE',''))
if 'direct-conv' in program_tags:
if kernel != '1' and kernel != '3' and kernel != '5':
return {'return':1, 'error':'direct-conv currently supports only kernels 1, 3 and 5 (you selected '+kernel+')'}
return {'return':1, 'error':'direct-conv only supports kernel=1,3,5 (you selected kernel={})'.format(kernel)}
elif 'winograd-conv' in program_tags:
if (kernel != '3' and kernel != '5') or stride != '1':
return {'return':1, 'error': 'winograd-conv only supports kernels 1 and 5 and stride=1 ' +
return {'return':1, 'error': 'winograd-conv only supports kernel=3,5, and stride=1 ' +
'(you selected kernel={}, stride={})'.format(kernel, stride)}

new_env = {}
Expand Down

0 comments on commit 2999c74

Please sign in to comment.