minor fix

apache · Dec 23, 2021 · 6ee5a39 · 6ee5a39
1 parent 8af8b30
commit 6ee5a39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/contrib/cutlass.py b/python/tvm/relay/op/contrib/cutlass.py
@@ -169,7 +169,7 @@ def check_conv2d_residual(call, binary_op):
     if get_root_call(lhs, "nn.conv2d") == conv2d and get_root_call(rhs, "nn.conv2d") == conv2d:
         return True
 
-    return all([x == y for (x, y) in zip(lhs.checked_type.shape, rhs.checked_type.shape)])
+    return all(x == y for (x, y) in zip(lhs.checked_type.shape, rhs.checked_type.shape))
 
 
 def partition_for_cutlass(mod, params=None):