[ROCm][CI] Disable async scheduling on ROCm for test_structured_output[meta-llama/Meta-Llama-3.1-8B-Instruct-xgrammar-auto-speculative_config9] (#32355)
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
This commit is contained in:
@@ -87,6 +87,10 @@ PARAMS_MODELS_TOKENIZER_MODE = [
|
||||
("Qwen/Qwen2.5-1.5B-Instruct", "auto"),
|
||||
]
|
||||
|
||||
platform_args = {}
|
||||
if current_platform.is_rocm():
|
||||
platform_args["async_scheduling"] = False
|
||||
|
||||
|
||||
class CarType(str, Enum):
|
||||
sedan = "sedan"
|
||||
@@ -134,6 +138,7 @@ def test_structured_output(
|
||||
load_format="auto" if not model_name.startswith("mistralai/") else "hf",
|
||||
config_format="auto" if not model_name.startswith("mistralai/") else "hf",
|
||||
speculative_config=speculative_config,
|
||||
**platform_args,
|
||||
)
|
||||
|
||||
#
|
||||
|
||||
Reference in New Issue
Block a user