diff --git a/setup.py b/setup.py index 3a443f95a..a90596d3d 100755 --- a/setup.py +++ b/setup.py @@ -217,7 +217,8 @@ def append_nvcc_threads(nvcc_extra_args): "nvcc": [ "-O3", "-std=c++17", - f"--offload-arch={os.getenv('HIP_ARCHITECTURES', 'native')}", + # Support multiple architectures via semicolon-separated HIP_ARCHITECTURES + *[f'--offload-arch={arch}' for arch in os.getenv('HIP_ARCHITECTURES', 'native').split(';')], "-U__CUDA_NO_HALF_OPERATORS__", "-U__CUDA_NO_HALF_CONVERSIONS__", "-fgpu-flush-denormals-to-zero",