diff --git a/tests/python/contrib/test_cutlass.py b/tests/python/contrib/test_cutlass.py index e041e9fe2725..585b42a21425 100644 --- a/tests/python/contrib/test_cutlass.py +++ b/tests/python/contrib/test_cutlass.py @@ -386,7 +386,7 @@ def test_conv2d(): sm=80, atol=1e-5, rtol=1e-5, - use_cudnn_ref=IC == 3, + use_cudnn_ref=(IC == 3), # The autotvm kernel has an accuracy issue with IC == 3 case run_benchmark=False, )