hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

sirius: fix cuda_arch flags (#39889)

Browse source
This commit is contained in:
Simon Pintarelli 2023-09-11 18:13:49 +02:00 committed by GitHub
parent 516f0461b8
commit d88ecf0af0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
var/spack/repos/builtin/packages/sirius/package.py
View file

@ -342,8 +342,10 @@ def cmake_args(self):
)
# Make SIRIUS handle it
elif "@6:7.4.3" in spec:
args.append(self.define("CMAKE_CUDA_ARCH", ";".join(cuda_arch)))
else:
args.append(self.define("CUDA_ARCH", ";".join(cuda_arch)))
args.append(self.define("CMAKE_CUDA_ARCHITECTURES", ";".join(cuda_arch)))
if "+rocm" in spec:
archs = ",".join(self.spec.variants["amdgpu_target"].value)