Skip failing inference test which requires change to PT core (#640)

pytorch · Aug 8, 2024 · 5f35645 · 5f35645
1 parent 0a3b328
commit 5f35645
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/float8/test_inference_flows.py b/test/float8/test_inference_flows.py
@@ -267,6 +267,7 @@ def setup_mock(self):
         not torch.cuda.is_available() or not is_H100,
         "CUDA not available or on non H100 machine",
     )
+    @unittest.skip("Pytorch needs a fix to ensure codegen maintains stride order")
     def test_fp8_export(self):
         export_model = FeedForward().to("cuda")
         quant_config = QuantConfig(ActivationCasting.DYNAMIC)