[AMD] Clear pre-built AITER kernels and warmup to prevent segfaults and test timeouts (#15318)

This commit is contained in:
sunxxuns
2025-12-18 14:00:09 -08:00
committed by GitHub
parent e0026f7c92
commit e0963a6cb1
4 changed files with 142 additions and 3 deletions
+2 -1
View File
@@ -6,7 +6,7 @@ from typing import List
import torch
from sglang.srt.utils import kill_process_tree
from sglang.srt.utils import is_hip, kill_process_tree
from sglang.test.runners import DEFAULT_PROMPTS, SRTRunner, check_close_model_outputs
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST,
@@ -66,6 +66,7 @@ class TestTransformersFallbackEndpoint(CustomTestCase):
self.assertGreater(metrics["accuracy"], self.gsm8k_lower_bound)
@unittest.skipIf(is_hip(), "TorchAO int4wo quantization is not supported on AMD GPUs")
class TestTransformersFallbackTorchAO(TestTransformersFallbackEndpoint):
@classmethod
def setUpClass(cls):