@@ -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):
|
||||
|
Reference in New Issue
Block a user