[TOPI][CUDA] minor change on assert statement in conv2d_NCHWc_int8.cuda (#8554)

zotanika · web-flow · commit a9933211a926 · 2021-07-29T19:33:44.000+09:00
* [TOPI][CUDA] minor change on assert statement

* [TOPI][CUDA] reformatting
diff --git a/python/tvm/topi/cuda/conv2d_int8.py b/python/tvm/topi/cuda/conv2d_int8.py
@@ -98,9 +98,9 @@ def conv2d_NCHWc_int8(cfg, data, kernel, stride, padding, dilation, layout, out_
         )
 
         out_channels, in_channels, kernel_h, kernel_w = get_const_tuple(kernel.shape)
-        assert out_channels % 4 == 0, "Number of output channels should be multiple of {}".format(
-            oc_block_factor
-        )
+        assert (
+            out_channels % oc_block_factor == 0
+        ), "Number of output channels should be multiple of {}".format(oc_block_factor)
         packed_kernel = te.compute(
             (
                 out_channels // oc_block_factor,

Original file line number	Diff line number	Diff line change
`@@ -98,9 +98,9 @@ def conv2d_NCHWc_int8(cfg, data, kernel, stride, padding, dilation, layout, out_`
`98`	`98`	`)`
`99`	`99`
`100`	`100`	`out_channels, in_channels, kernel_h, kernel_w = get_const_tuple(kernel.shape)`
`101`		`- assert out_channels % 4 == 0, "Number of output channels should be multiple of {}".format(`
`102`		`- oc_block_factor`
`103`		`- )`
	`101`	`+ assert (`
	`102`	`+ out_channels % oc_block_factor == 0`
	`103`	`+ ), "Number of output channels should be multiple of {}".format(oc_block_factor)`
`104`	`104`	`packed_kernel = te.compute(`
`105`	`105`	`(`
`106`	`106`	`out_channels // oc_block_factor,`