From e29d3aa5b42f88968da24b619fd4552f561bb5e1 Mon Sep 17 00:00:00 2001 From: Albert Malewski Date: Thu, 14 Aug 2025 14:51:15 +0000 Subject: [PATCH] [release/2.7] Added missing MI350_ARCH constant value --- torch/testing/_internal/common_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch/testing/_internal/common_utils.py b/torch/testing/_internal/common_utils.py index 7118ce3a957dc..d3d30d06cbdf6 100644 --- a/torch/testing/_internal/common_utils.py +++ b/torch/testing/_internal/common_utils.py @@ -104,6 +104,7 @@ MI300_ARCH = ("gfx940", "gfx941", "gfx942") +MI350_ARCH = ("gfx950") NAVI_ARCH = ("gfx1030", "gfx1100", "gfx1101", "gfx1200", "gfx1201") NAVI3_ARCH = ("gfx1100", "gfx1101") NAVI4_ARCH = ("gfx1200", "gfx1201")