summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSreenivasa Murthy Kolam <67086238+srekolam@users.noreply.github.com>2021-08-13 12:08:25 -0700
committerGitHub <noreply@github.com>2021-08-13 19:08:25 +0000
commit1c7aa1261534533ef789ed9a9d4a96f802aae0f6 (patch)
tree31bf8b330a63614ad045f07bb0dcbf57f94d5c03
parentf0875b2fef8e495809854141785c1fc7f5d86407 (diff)
downloadspack-1c7aa1261534533ef789ed9a9d4a96f802aae0f6.tar.gz
spack-1c7aa1261534533ef789ed9a9d4a96f802aae0f6.tar.bz2
spack-1c7aa1261534533ef789ed9a9d4a96f802aae0f6.tar.xz
spack-1c7aa1261534533ef789ed9a9d4a96f802aae0f6.zip
cleanup opt-rocm references when using spack (#25264)
Co-authored-by: Harmen Stoppels <harmenstoppels@gmail.com>
-rw-r--r--var/spack/repos/builtin/packages/hipblas/package.py3
-rw-r--r--var/spack/repos/builtin/packages/hipfft/package.py2
-rw-r--r--var/spack/repos/builtin/packages/hipsparse/package.py2
3 files changed, 7 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/hipblas/package.py b/var/spack/repos/builtin/packages/hipblas/package.py
index 86aee29b8c..4836e47699 100644
--- a/var/spack/repos/builtin/packages/hipblas/package.py
+++ b/var/spack/repos/builtin/packages/hipblas/package.py
@@ -32,9 +32,12 @@ class Hipblas(CMakePackage):
depends_on('rocsolver@' + ver, when='@' + ver)
depends_on('rocblas@' + ver, type='link', when='@' + ver)
depends_on('comgr@' + ver, type='build', when='@' + ver)
+ depends_on('rocm-cmake@' + ver, type='build', when='@' + ver)
def cmake_args(self):
args = [
+ # Make sure find_package(HIP) finds the module.
+ self.define('CMAKE_MODULE_PATH', self.spec['hip'].prefix.cmake),
self.define('BUILD_CLIENTS_SAMPLES', 'OFF'),
self.define('BUILD_CLIENTS_TESTS', 'OFF')
]
diff --git a/var/spack/repos/builtin/packages/hipfft/package.py b/var/spack/repos/builtin/packages/hipfft/package.py
index 3359af11d4..69f9514edc 100644
--- a/var/spack/repos/builtin/packages/hipfft/package.py
+++ b/var/spack/repos/builtin/packages/hipfft/package.py
@@ -38,6 +38,8 @@ class Hipfft(CMakePackage):
def cmake_args(self):
args = [
+ # Make sure find_package(HIP) finds the module.
+ self.define('CMAKE_MODULE_PATH', self.spec['hip'].prefix.cmake),
self.define('BUILD_CLIENTS_SAMPLES', 'OFF')
]
diff --git a/var/spack/repos/builtin/packages/hipsparse/package.py b/var/spack/repos/builtin/packages/hipsparse/package.py
index afc1c0fe39..c939b686ed 100644
--- a/var/spack/repos/builtin/packages/hipsparse/package.py
+++ b/var/spack/repos/builtin/packages/hipsparse/package.py
@@ -43,6 +43,8 @@ class Hipsparse(CMakePackage):
def cmake_args(self):
args = [
+ # Make sure find_package(HIP) finds the module.
+ self.define('CMAKE_MODULE_PATH', self.spec['hip'].prefix.cmake),
self.define('CMAKE_CXX_STANDARD', '14'),
self.define('BUILD_CLIENTS_SAMPLES', 'OFF'),
self.define('BUILD_CLIENTS_TESTS', 'OFF'),