diff --git a/setup.py b/setup.py index ea58a1a49e7e3..8fcb86394f76d 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)