summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/llvm-amdgpu/package.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/llvm-amdgpu/package.py b/var/spack/repos/builtin/packages/llvm-amdgpu/package.py
index 3e39a4445b..4fec91e9ea 100644
--- a/var/spack/repos/builtin/packages/llvm-amdgpu/package.py
+++ b/var/spack/repos/builtin/packages/llvm-amdgpu/package.py
@@ -222,12 +222,16 @@ class LlvmAmdgpu(CMakePackage, CompilerPackage):
self.define("LIBCXXABI_ENABLE_STATIC", "ON"),
self.define("LIBCXXABI_INSTALL_STATIC_LIBRARY", "OFF"),
self.define("LLVM_ENABLE_RTTI", "ON"),
- self.define("LLVM_TARGETS_TO_BUILD", "AMDGPU;X86"),
self.define("LLVM_AMDGPU_ALLOW_NPI_TARGETS", "ON"),
self.define("PACKAGE_VENDOR", "AMD"),
self.define("CLANG_ENABLE_AMDCLANG", "ON"),
]
+ if self.spec.target.family == "aarch64":
+ args.append(self.define("LLVM_TARGETS_TO_BUILD", "AMDGPU;AArch64"))
+ else:
+ args.append(self.define("LLVM_TARGETS_TO_BUILD", "AMDGPU;X86"))
+
# Enable rocm-device-libs as a external project
if self.spec.satisfies("+rocm-device-libs"):
if self.spec.satisfies("@:6.0"):