From 870823c6d54114896aa9db91eb72c132cdef762f Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 10 Dec 2021 12:54:38 +0900 Subject: [PATCH] add comment on IC == 3 case --- 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 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, )