Skip to content

Commit bbf128a

Browse files
committed
more pylint fixes
1 parent 1ba47de commit bbf128a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

topi/python/topi/cuda/conv2d.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,10 @@ def conv2d_cuda(cfg, data, kernel, strides, padding, dilation, layout='NCHW', ou
104104
if cfg.template_key == 'winograd':
105105
return winograd_cuda(cfg, data, kernel, strides, padding, dilation, layout, out_dtype,
106106
pre_computed=False)
107-
if cfg.template_key == 'int8' :
108-
if (data.dtype=='int8' or data.dtype=='uint8'):
109-
return conv2d_NCHWc_int8(cfg, data, kernel, strides, padding, dilation, layout, out_dtype)
107+
if cfg.template_key == 'int8':
108+
if (data.dtype == 'int8' or data.dtype == 'uint8'):
109+
return conv2d_NCHWc_int8(
110+
cfg, data, kernel, strides, padding, dilation, layout, out_dtype)
110111

111112
if layout == 'NCHW':
112113
return nn.conv2d_nchw(data, kernel, strides, padding, dilation, out_dtype)

0 commit comments

Comments
 (0)