19 lines
589 B
Python
19 lines
589 B
Python
# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
|
|
from spack.package import *
|
|
from spack.pkg.builtin.cuda import Cuda as SpackCuda
|
|
|
|
|
|
class Cuda(SpackCuda):
|
|
|
|
def setup_run_environment(self, env):
|
|
super().setup_run_environment(self, env)
|
|
|
|
if '%nvhpc' in self.spec:
|
|
env.prepend_path("LD_LIBRARY_PATH", self.prefix.lib64)
|
|
env.prepend_path("LD_LIBRARY_PATH", join_path(self.prefix, '../math_libs/lib64'))
|