We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 386ea5c commit 73740c4Copy full SHA for 73740c4
python/tvm/relay/op/contrib/cutlass.py
@@ -106,7 +106,7 @@ def check_conv2d(call):
106
return False
107
IC = data.shape[3]
108
OC = weight.shape[0]
109
- return not is_depthwise_conv2d(IC, OC, call.attrs.groups)
+ return not is_depthwise_conv2d(IC, OC, conv2d.attrs.groups)
110
111
112
def partition_for_cutlass(mod):
0 commit comments