EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[CI fix] test skipping modelopt on AMD (#4677)

Browse Source
This commit is contained in:
Adarsh Shirawalmath
2025-03-23 01:06:02 +05:30
committed by GitHub
parent fb8886037c
commit a2cc62a6db
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/srt/test_eval_fp8_accuracy.py
View File

@@ -1,6 +1,8 @@
import unittest import unittest
from types import SimpleNamespace from types import SimpleNamespace
import torch
from sglang.srt.utils import kill_process_tree from sglang.srt.utils import kill_process_tree
from sglang.test.run_eval import run_eval from sglang.test.run_eval import run_eval
from sglang.test.test_utils import ( from sglang.test.test_utils import (
@@ -135,6 +137,9 @@ class TestEvalFP8ModelOptQuantAccuracy(unittest.TestCase):
finally: finally:
kill_process_tree(process.pid) kill_process_tree(process.pid)
@unittest.skipIf(
torch.version.hip is not None, "modelopt quantization unsupported on ROCm"
)
def test_mmlu_offline_only(self): def test_mmlu_offline_only(self):
"""Test with offline quantization only.""" """Test with offline quantization only."""
self._run_test( self._run_test(