update test_std_correction with skip on MLU

DeepLink-org · Jul 31, 2024 · bb33e20 · bb33e20
1 parent bf93ec6
commit bb33e20
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dipu/tests/python/unittests/test_mean_std.py b/dipu/tests/python/unittests/test_mean_std.py
@@ -1,7 +1,7 @@
 # Copyright (c) 2023, DeepLink.
 import torch
 import torch_dipu
-from torch_dipu.testing._internal.common_utils import TestCase, run_tests
+from torch_dipu.testing._internal.common_utils import TestCase, run_tests, skipOn
 
 
 class TestMeanStd(TestCase):
@@ -60,6 +60,9 @@ def test_std(self):
                 rtol=1e-3,
             )
         )
+
+    @skipOn("MLU", "camb does not support this type")
+    def test_std_correction(self):
         self.assertTrue(
             torch.allclose(
                 torch.std(self.a, dim=-1, correction=20).cpu(),