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:
|
||||
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.
|
||||
|
|
Loading…
Reference in a new issue