"test/gemm_xdl/gemm_int8.cpp" did not exist on "0c79af12e882c29c1f5a2895e6f749cdee9e15b7"
test_torch_compile.py 1018 Bytes
Newer Older
1
2
3
4
5
import unittest
from types import SimpleNamespace

from sglang.srt.utils import kill_child_process
from sglang.test.run_eval import run_eval
6
7
8
9
10
from sglang.test.test_utils import (
    DEFAULT_MODEL_NAME_FOR_TEST,
    DEFAULT_URL_FOR_TEST,
    popen_launch_server,
)
11
12


13
class TestTorchCompile(unittest.TestCase):
14
15
16

    @classmethod
    def setUpClass(cls):
Ying Sheng's avatar
Ying Sheng committed
17
        cls.model = DEFAULT_MODEL_NAME_FOR_TEST
18
        cls.base_url = DEFAULT_URL_FOR_TEST
19
20
21
22
23
24
25
26
27
28
29
30
31
        cls.process = popen_launch_server(
            cls.model, cls.base_url, timeout=300, other_args=["--enable-torch-compile"]
        )

    @classmethod
    def tearDownClass(cls):
        kill_child_process(cls.process.pid)

    def test_mmlu(self):
        args = SimpleNamespace(
            base_url=self.base_url,
            model=self.model,
            eval_name="mmlu",
32
33
            num_examples=32,
            num_threads=32,
34
35
36
        )

        metrics = run_eval(args)
37
        assert metrics["score"] >= 0.6
38
39
40


if __name__ == "__main__":
Lianmin Zheng's avatar
Lianmin Zheng committed
41
    unittest.main()