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

lbann, aluminum, hydrogen,dihydrogen: explicit flags to pass the AMD GPU architecture into CMake (#31326)

Browse Source
This commit is contained in:
Brian Van Essen 2022-06-28 00:45:00 -07:00 committed by GitHub
parent 899dfdb416
commit 8965d8f661
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/aluminum/package.py
View File

@ -126,5 +126,10 @@ def cmake_args(self):
'-DHIP_HIPCC_FLAGS=--amdgpu-target={0}'
' -g -fsized-deallocation -fPIC -std=c++17'.format(arch_str)
)
args.extend([
'-DCMAKE_HIP_ARCHITECTURES=%s' % arch_str,
'-DAMDGPU_TARGETS=%s' % arch_str,
'-DGPU_TARGETS=%s' % arch_str,
])
return args

5
var/spack/repos/builtin/packages/dihydrogen/package.py
View File

@ -209,6 +209,11 @@ def cmake_args(self):
'-DHIP_HIPCC_FLAGS=--amdgpu-target={0}'
' -g -fsized-deallocation -fPIC -std=c++17'.format(arch_str)
)
args.extend([
'-DCMAKE_HIP_ARCHITECTURES=%s' % arch_str,
'-DAMDGPU_TARGETS=%s' % arch_str,
'-DGPU_TARGETS=%s' % arch_str,
])
if self.spec.satisfies('^essl'):
# IF IBM ESSL is used it needs help finding the proper LAPACK libraries

5
var/spack/repos/builtin/packages/hydrogen/package.py
View File

@ -201,6 +201,11 @@ def cmake_args(self):
' -g -fsized-deallocation -fPIC {1}'
' -std=c++17'.format(arch_str, cxxflags_str)
)
args.extend([
'-DCMAKE_HIP_ARCHITECTURES=%s' % arch_str,
'-DAMDGPU_TARGETS=%s' % arch_str,
'-DGPU_TARGETS=%s' % arch_str,
])
# Add support for OS X to find OpenMP (LLVM installed via brew)
if self.spec.satisfies('%clang +openmp platform=darwin'):

5
var/spack/repos/builtin/packages/lbann/package.py
View File

@ -411,6 +411,11 @@ def cmake_args(self):
' -g -fsized-deallocation -fPIC -std=c++17 {1}'.format(
arch_str, cxxflags_str)
)
args.extend([
'-DCMAKE_HIP_ARCHITECTURES=%s' % arch_str,
'-DAMDGPU_TARGETS=%s' % arch_str,
'-DGPU_TARGETS=%s' % arch_str,
])
# IF IBM ESSL is used it needs help finding the proper LAPACK libraries
if self.spec.satisfies('^essl'):