diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/openmm/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/openmm/package.py | 22 |
1 files changed, 19 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/openmm/package.py b/var/spack/repos/builtin/packages/openmm/package.py index aa7c1c1b63..3b8329fdea 100644 --- a/var/spack/repos/builtin/packages/openmm/package.py +++ b/var/spack/repos/builtin/packages/openmm/package.py @@ -3,8 +3,6 @@ # # SPDX-License-Identifier: (Apache-2.0 OR MIT) -import os - from spack.package import * @@ -49,8 +47,26 @@ class Openmm(CMakePackage, CudaPackage): "wrappers/python/CMakeLists.txt", ) + def setup_build_environment(self, env): + spec = self.spec + if "+cuda" in spec: + env.set("OPENMM_CUDA_COMPILER", self.spec["cuda"].prefix.bin.nvcc) + env.set("CUDA_HOST_COMPILER", self.compiler.cxx) + def setup_run_environment(self, env): spec = self.spec if "+cuda" in spec: env.set("OPENMM_CUDA_COMPILER", self.spec["cuda"].prefix.bin.nvcc) - env.prepend_path("PATH", os.path.dirname(self.compiler.cc)) + env.set("CUDA_HOST_COMPILER", self.compiler.cxx) + + def setup_dependent_run_environment(self, env, dependent_spec): + spec = self.spec + if "+cuda" in spec: + env.set("OPENMM_CUDA_COMPILER", self.spec["cuda"].prefix.bin.nvcc) + env.set("CUDA_HOST_COMPILER", self.compiler.cxx) + + def setup_dependent_build_environment(self, env, dependent_spec): + spec = self.spec + if "+cuda" in spec: + env.set("OPENMM_CUDA_COMPILER", self.spec["cuda"].prefix.bin.nvcc) + env.set("CUDA_HOST_COMPILER", self.compiler.cxx) |