Unverified Commit 773d7073 authored by Micah Williamson's avatar Micah Williamson Committed by GitHub
Browse files

[ROCm][CI] Disable async scheduling on ROCm for...


[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: default avatarMicah Williamson <micah.williamson@amd.com>
parent edadca10
......@@ -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,
)
#
......
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