Unverified Commit a2cc62a6 authored by Adarsh Shirawalmath's avatar Adarsh Shirawalmath Committed by GitHub
Browse files

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

parent fb888603
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): ...@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment