correct skip

Xilinx · May 29, 2024 · ac5e58c · ac5e58c
1 parent d762c99
commit ac5e58c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/brevitas/export/test_onnx_fp8.py b/tests/brevitas/export/test_onnx_fp8.py
@@ -10,10 +10,10 @@
 import brevitas.nn as qnn
 from brevitas.quant.experimental.float_quant_ocp import Fp8e4m3OCPActPerTensorFloat
 from brevitas.quant.experimental.float_quant_ocp import Fp8e4m3OCPWeightPerTensorFloat
-from tests.marker import jit_disabled_for_mock
+from tests.marker import jit_disabled_for_export
 
 
-@jit_disabled_for_mock()
+@jit_disabled_for_export()
 def test_simple_fp8_export():
     if torch_version < version.parse('2.1.0'):
         pytest.skip(f"OCP FP8 types not supported by {torch_version}")
@@ -23,7 +23,7 @@ def test_simple_fp8_export():
     assert True
 
 
-@jit_disabled_for_mock()
+@jit_disabled_for_export()
 def test_fp8_export_activation():
     if torch_version < version.parse('2.1.0'):
         pytest.skip(f"OCP FP8 types not supported by {torch_version}")
@@ -33,7 +33,7 @@ def test_fp8_export_activation():
     assert True
 
 
-@jit_disabled_for_mock()
+@jit_disabled_for_export()
 def test_fp8_export_export_activation():
     if torch_version < version.parse('2.1.0'):
         pytest.skip(f"OCP FP8 types not supported by {torch_version}")