Fix self.spec reference (#47610)

@teaguesterling
This commit is contained in:
wspear 2024-11-18 10:37:53 -08:00 committed by GitHub
parent 15b3ff2a0a
commit 1c75d07f05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -44,7 +44,7 @@ def setup_build_environment(self, env):
cuda_prefix = self.spec["cuda"].prefix
env.set("CUDACXX", cuda_prefix.bin.nvcc)
env.set("CUDA_LIB_DIR", cuda_prefix.lib)
env.set("CMAKE_CUDA_ARCHITECTURES", spec.variants["cuda_arch"].value)
env.set("CMAKE_CUDA_ARCHITECTURES", self.spec.variants["cuda_arch"].value)
@property
def generate_args(self):