zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

hiop +rocm: add amdgpu_target for deps (#30184)

Browse Source 
* hiop +rocm: add amdgpu_target for deps

* Update var/spack/repos/builtin/packages/hiop/package.py

Perfect! Thanks!

Co-authored-by: Asher Mancinelli <ashermancinelli@gmail.com>

* don't use f-string, not compatible with py 2

Co-authored-by: Asher Mancinelli <ashermancinelli@gmail.com>
This commit is contained in:
eugeneswalker 2022-04-22 16:26:38 -07:00 committed by GitHub
parent 78d87c71f6
commit 58ee164777
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
var/spack/repos/builtin/packages/hiop/package.py
View File

@ -66,7 +66,12 @@ class Hiop(CMakePackage, CudaPackage, ROCmPackage):
depends_on('mpi', when='+mpi')
depends_on('magma+cuda', when='+cuda')
depends_on('magma+rocm', when='+rocm')
for arch in ROCmPackage.amdgpu_targets:
rocm_dep = "+rocm amdgpu_target={0}".format(arch)
depends_on("magma {0}".format(rocm_dep), when=rocm_dep)
depends_on("raja {0}".format(rocm_dep), when="+raja {0}".format(rocm_dep))
depends_on("umpire {0}".format(rocm_dep), when="+raja {0}".format(rocm_dep))
# Depends on Magma when +rocm or +cuda
magma_ver_constraints = (
@ -82,10 +87,8 @@ class Hiop(CMakePackage, CudaPackage, ROCmPackage):
depends_on('raja+openmp', when='+raja~cuda~rocm')
depends_on('raja@0.14.0:', when='@0.5.0:+raja')
depends_on('raja+cuda', when='+raja+cuda')
depends_on('raja+rocm', when='+raja+rocm')
depends_on('umpire', when='+raja')
depends_on('umpire+cuda~shared', when='+raja+cuda')
depends_on('umpire+rocm', when='+raja+rocm')
depends_on('umpire@6.0.0:', when='@0.5.0:+raja')
depends_on('hip', when='+rocm')
depends_on('hipblas', when='+rocm')