fixed for sm75

masahi · Feb 6, 2022 · a145850 · a145850
1 parent 6151506
commit a145850
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/contrib/test_cutlass.py b/tests/python/contrib/test_cutlass.py
@@ -518,7 +518,7 @@ def verify_conv2d_common(
 ):
     if not has_cutlass():
         return
-    if sm < 80 and data_dtype == "float32":
+    if sm < 80 and inputs[0].dtype == "float32":
         return
 
     mod_nchw = tvm.IRModule.from_expr(expr_nchw)