update unit test in AMD CI (#3366)

test/srt/test_mla.py

@@ -1,6 +1,8 @@
 import unittest
 from types import SimpleNamespace
 
+import torch
+
 from sglang.srt.utils import kill_process_tree
 from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
 from sglang.test.run_eval import run_eval
@@ -58,16 +60,14 @@ class TestDeepseekV3(unittest.TestCase):
     def setUpClass(cls):
         cls.model = "lmzheng/sglang-ci-dsv3-test"
         cls.base_url = DEFAULT_URL_FOR_TEST
+        other_args = ["--trust-remote-code"]
+        if torch.cuda.is_available() and torch.version.cuda:
+            other_args.extend(["--enable-torch-compile", "--cuda-graph-max-bs", "2"])
         cls.process = popen_launch_server(
             cls.model,
             cls.base_url,
             timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
-            other_args=[
-                "--trust-remote-code",
-                "--enable-torch-compile",
-                "--cuda-graph-max-bs",
-                "2",
-            ],
+            other_args=other_args,
         )
 
     @classmethod