From 264017a2bf030f060ebad91eb9be9b4e0033edb9 Mon Sep 17 00:00:00 2001 From: James Whedbee Date: Mon, 19 Feb 2024 19:58:59 -0600 Subject: [PATCH] [ROCm] include gfx908 as supported (#2792) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index ea58a1a49e7e..8fcb86394f76 100644 --- a/setup.py +++ b/setup.py @@ -24,7 +24,7 @@ MAIN_CUDA_VERSION = "12.1" # Supported NVIDIA GPU architectures. NVIDIA_SUPPORTED_ARCHS = {"7.0", "7.5", "8.0", "8.6", "8.9", "9.0"} -ROCM_SUPPORTED_ARCHS = {"gfx90a", "gfx942", "gfx1100"} +ROCM_SUPPORTED_ARCHS = {"gfx908", "gfx90a", "gfx942", "gfx1100"} # SUPPORTED_ARCHS = NVIDIA_SUPPORTED_ARCHS.union(ROCM_SUPPORTED_ARCHS)