From a2cc62a6db68f04826f31fa2f2749026b60c7385 Mon Sep 17 00:00:00 2001 From: Adarsh Shirawalmath <114558126+adarshxs@users.noreply.github.com> Date: Sun, 23 Mar 2025 01:06:02 +0530 Subject: [PATCH] [CI fix] test skipping modelopt on AMD (#4677) --- test/srt/test_eval_fp8_accuracy.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/test/srt/test_eval_fp8_accuracy.py b/test/srt/test_eval_fp8_accuracy.py index d36216dd2..9431d14d3 100644 --- a/test/srt/test_eval_fp8_accuracy.py +++ b/test/srt/test_eval_fp8_accuracy.py @@ -1,6 +1,8 @@ import unittest from types import SimpleNamespace +import torch + from sglang.srt.utils import kill_process_tree from sglang.test.run_eval import run_eval from sglang.test.test_utils import ( @@ -135,6 +137,9 @@ class TestEvalFP8ModelOptQuantAccuracy(unittest.TestCase): finally: kill_process_tree(process.pid) + @unittest.skipIf( + torch.version.hip is not None, "modelopt quantization unsupported on ROCm" + ) def test_mmlu_offline_only(self): """Test with offline quantization only.""" self._run_test(