dihydrogen: propagate cuda_arch to nvshmem (#30886)

This commit is contained in:
Brian Van Essen 2022-06-06 14:38:58 +02:00 committed by GitHub
parent 8ef937032c
commit 926e311f3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -72,6 +72,7 @@ class Dihydrogen(CMakePackage, CudaPackage, ROCmPackage):
for arch in CudaPackage.cuda_arch_values:
depends_on('aluminum cuda_arch=%s' % arch, when='+al +cuda cuda_arch=%s' % arch)
depends_on('nvshmem cuda_arch=%s' % arch, when='+nvshmem +cuda cuda_arch=%s' % arch)
# variants +rocm and amdgpu_targets are not automatically passed to
# dependencies, so do it manually.