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

View File

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