add comment on IC == 3 case

ylc · Dec 11, 2021 · 870823c · 870823c
1 parent 6b780db
commit 870823c
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
@@ -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,
         )