From a14585020151d0e09bb9bac549285dceb13e55e1 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sun, 6 Feb 2022 14:46:16 +0900 Subject: [PATCH] fixed for sm75 --- tests/python/contrib/test_cutlass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/contrib/test_cutlass.py b/tests/python/contrib/test_cutlass.py index 3a7bdf43fd810..ef55c74dc3a5e 100644 --- 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)