dihydrogen: propagate cuda_arch
to nvshmem (#30886)
This commit is contained in:
parent
8ef937032c
commit
926e311f3c
1 changed files with 1 additions and 0 deletions
|
@ -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.
|
||||||
|
|
Loading…
Reference in a new issue