diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/ginkgo/package.py | 44 |
1 files changed, 41 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/ginkgo/package.py b/var/spack/repos/builtin/packages/ginkgo/package.py index d0ee354256..f72fed7319 100644 --- a/var/spack/repos/builtin/packages/ginkgo/package.py +++ b/var/spack/repos/builtin/packages/ginkgo/package.py @@ -18,6 +18,7 @@ class Ginkgo(CMakePackage, CudaPackage): version('develop', branch='develop') version('master', branch='master') + version('1.3.0', commit='4678668c66f634169def81620a85c9a20b7cec78') # v1.3.0 version('1.2.0', commit='b4be2be961fd5db45c3d02b5e004d73550722e31') # v1.2.0 version('1.1.1', commit='08d2c5200d3c78015ac8a4fd488bafe1e4240cf5') # v1.1.1 version('1.1.0', commit='b9bec8225442b3eb2a85a870efa112ab767a17fb') # v1.1.0 @@ -30,15 +31,42 @@ class Ginkgo(CMakePackage, CudaPackage): variant('build_type', default='Release', description='The build type to build', values=('Debug', 'Release')) + variant('hip', default=False, description='Compile Ginkgo with HIP support') depends_on('cmake@3.9:', type='build') depends_on('cuda@9:', when='+cuda') + depends_on('hip', when='+hip') + depends_on('hipsparse', type="link", when='+hip') + depends_on('hipblas', type="link", when='+hip') + depends_on('rocrand', type="link", when='@develop+hip') + depends_on('rocthrust', type="build", when='+hip') + + # Somehow, these dependencies not propagated by the HIP stack? + depends_on('rocm-device-libs', type="link", when='+hip') + depends_on('comgr', type="link", when='+hip') + conflicts('%gcc@:5.2.9') + conflicts("+hip", when="@:1.1.1") + # The HIP packages from spack doen't seem to work well with CUDA + # backend for now, so disable HIP with CUDA backend. + conflicts("+cuda", when="+hip") def cmake_args(self): + # Check that the have the correct C++ standard is available + if self.spec.satisfies('@:1.2.0'): + try: + self.compiler.cxx11_flag + except UnsupportedCompilerFlag: + InstallError('Ginkgo requires a C++11-compliant C++ compiler') + else: + try: + self.compiler.cxx14_flag + except UnsupportedCompilerFlag: + InstallError('Ginkgo requires a C++14-compliant C++ compiler') + spec = self.spec - return [ + args = [ '-DGINKGO_BUILD_CUDA=%s' % ('ON' if '+cuda' in spec else 'OFF'), '-DGINKGO_BUILD_OMP=%s' % ('ON' if '+openmp' in spec else 'OFF'), '-DBUILD_SHARED_LIBS=%s' % ('ON' if '+shared' in spec else 'OFF'), @@ -46,8 +74,7 @@ class Ginkgo(CMakePackage, CudaPackage): 'ON' if '+full_optimizations' in spec else 'OFF'), '-DGINKGO_DEVEL_TOOLS=%s' % ( 'ON' if '+develtools' in spec else 'OFF'), - # Drop HIP support for now - '-DGINKGO_BUILD_HIP=OFF', + '-DGINKGO_BUILD_HIP=%s' % ('ON' if '+hip' in spec else 'OFF'), # As we are not exposing benchmarks, examples, tests nor doc # as part of the installation, disable building them altogether. '-DGINKGO_BUILD_BENCHMARKS=OFF', @@ -55,3 +82,14 @@ class Ginkgo(CMakePackage, CudaPackage): '-DGINKGO_BUILD_EXAMPLES=OFF', '-DGINKGO_BUILD_TESTS=OFF' ] + if '+hip' in spec: + args.append('-DHIP_PATH={0}'. format(spec['hip'].prefix)) + args.append('-DHIP_CLANG_PATH={0}/bin'. + format(spec['llvm-amdgpu'].prefix)) + args.append('-DHIP_CLANG_INCLUDE_PATH={0}/include'. + format(spec['llvm-amdgpu'].prefix)) + args.append('-DHIPSPARSE_PATH={0}'. + format(spec['hipsparse'].prefix)) + args.append('-DHIPBLAS_PATH={0}'. + format(spec['hipblas'].prefix)) + return args |