skip test_softmax_bf16_large

microsoft · Aug 8, 2024 · b255667 · b255667
1 parent 080f05b
commit b255667
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/orttraining/orttraining/test/python/orttraining_test_ortmodule_onnx_ops.py b/orttraining/orttraining/test/python/orttraining_test_ortmodule_onnx_ops.py
@@ -150,6 +150,8 @@ def test_onnx_ops(self):
 
     @unittest.skipIf(not torch.cuda.is_bf16_supported(), "Test requires CUDA and BF16 support")
     def test_softmax_bf16_large(self):
+        raise unittest.SkipTest("Temporarily disabled pending investigation")
+
         if torch.version.cuda is None:
             # Only run this test when CUDA is available, as on ROCm BF16 is not supported by MIOpen.
             return