[TOPI][CUDA] minor change on assert statement in conv2d_NCHWc_int8.cu…

…da (#8554) * [TOPI][CUDA] minor change on assert statement * [TOPI][CUDA] reformatting
apache · Jul 29, 2021 · a993321 · a993321
1 parent b6b8269
commit a993321
Showing 1 changed file with 3 additions and 3 deletions.
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,