From 40a30fa10920597e7dfdfc8f617fb8225954e6e7 Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Mon, 20 Apr 2026 18:24:27 -0500 Subject: [PATCH] [ROCm][CI] Added MI300 queues in python pipeline generator Signed-off-by: Andreas Karatzas --- buildkite/pipeline_generator/buildkite_step.py | 4 ++++ buildkite/pipeline_generator/constants.py | 8 ++++++++ 2 files changed, 12 insertions(+) diff --git a/buildkite/pipeline_generator/buildkite_step.py b/buildkite/pipeline_generator/buildkite_step.py index 4b31fd0b..afa4a6cf 100644 --- a/buildkite/pipeline_generator/buildkite_step.py +++ b/buildkite/pipeline_generator/buildkite_step.py @@ -363,6 +363,10 @@ def _create_amd_mirror_step(step: Step, original_commands: List[str], amd: Dict[ DeviceType.AMD_MI250_2: AgentQueue.AMD_MI250_2, DeviceType.AMD_MI250_4: AgentQueue.AMD_MI250_4, DeviceType.AMD_MI250_8: AgentQueue.AMD_MI250_8, + DeviceType.AMD_MI300_1: AgentQueue.AMD_MI300_1, + DeviceType.AMD_MI300_2: AgentQueue.AMD_MI300_2, + DeviceType.AMD_MI300_4: AgentQueue.AMD_MI300_4, + DeviceType.AMD_MI300_8: AgentQueue.AMD_MI300_8, DeviceType.AMD_MI325_1: AgentQueue.AMD_MI325_1, DeviceType.AMD_MI325_2: AgentQueue.AMD_MI325_2, DeviceType.AMD_MI325_4: AgentQueue.AMD_MI325_4, diff --git a/buildkite/pipeline_generator/constants.py b/buildkite/pipeline_generator/constants.py index e4b979b9..b0c865f1 100644 --- a/buildkite/pipeline_generator/constants.py +++ b/buildkite/pipeline_generator/constants.py @@ -21,6 +21,10 @@ class DeviceType(str, Enum): AMD_MI250_2 = "mi250_2" AMD_MI250_4 = "mi250_4" AMD_MI250_8 = "mi250_8" + AMD_MI300_1 = "mi300_1" + AMD_MI300_2 = "mi300_2" + AMD_MI300_4 = "mi300_4" + AMD_MI300_8 = "mi300_8" AMD_MI325_1 = "mi325_1" AMD_MI325_2 = "mi325_2" AMD_MI325_4 = "mi325_4" @@ -59,6 +63,10 @@ class AgentQueue(str, Enum): AMD_MI250_2 = "amd_mi250_2" AMD_MI250_4 = "amd_mi250_4" AMD_MI250_8 = "amd_mi250_8" + AMD_MI300_1 = "amd_mi300_1" + AMD_MI300_2 = "amd_mi300_2" + AMD_MI300_4 = "amd_mi300_4" + AMD_MI300_8 = "amd_mi300_8" AMD_MI325_1 = "amd_mi325_1" AMD_MI325_2 = "amd_mi325_2" AMD_MI325_4 = "amd_mi325_4"