fixed conv2d check

masahi · masahi · commit 463574ce087c · 2021-12-12T17:28:38.000+09:00
diff --git a/python/tvm/relay/op/contrib/cutlass.py b/python/tvm/relay/op/contrib/cutlass.py
@@ -105,7 +105,7 @@ def check_conv2d(call):
         return False
     IC = data.shape[3]
     OC = weight.shape[0]
-    return not is_depthwise_conv2d(IC, OC, call.attrs.groups)
+    return not is_depthwise_conv2d(IC, OC, conv2d.attrs.groups)
 
 
 def partition_for_cutlass(mod):
