fixed conv2d check

masahi · Dec 12, 2021 · 463574c · 463574c
1 parent 588c5ab
commit 463574c
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
@@ -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):