diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index ef717f8d1..76d2ce4a7 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -829,7 +829,7 @@ steps: - pytest -v -s tests/kernels/quantization/test_flashinfer_scaled_mm.py - pytest -v -s tests/kernels/quantization/test_flashinfer_nvfp4_scaled_mm.py - pytest -v -s tests/kernels/moe/test_nvfp4_moe.py - - pytest -v -s tests/kernels/moe/test_mxfp4_moe.py + - pytest -v -s tests/kernels/moe/test_ocp_mx_moe.py # Fusion - pytest -v -s tests/compile/test_fusion_all_reduce.py - pytest -v -s tests/compile/test_fusion_attn.py::test_attention_quant_pattern diff --git a/tests/quantization/test_quark.py b/tests/quantization/test_quark.py index 6b6de067d..1e65d9a99 100644 --- a/tests/quantization/test_quark.py +++ b/tests/quantization/test_quark.py @@ -4,7 +4,7 @@ Run `pytest tests/quantization/test_quark.py`. -See also `tests/kernels/moe/test_mxfp4_moe.py`. +See also `tests/kernels/moe/test_ocp_mx_moe.py`. """ import importlib.metadata