[Feature] fix test_deform_conv

open-mmlab · Jun 30, 2023 · 8fbe007 · 8fbe007
1 parent 7b9a492
commit 8fbe007
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_ops/test_deform_conv.py b/tests/test_ops/test_deform_conv.py
@@ -7,6 +7,9 @@
 
 from mmcv.utils import IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE
 
+if IS_MLU_AVAILABLE:
+    torch.backends.cnnl.allow_tf32 = False
+
 try:
     # If PyTorch version >= 1.6.0 and fp16 is enabled, torch.cuda.amp.autocast
     # would be imported and used; we should test if our modules support it.