598c233f78
* change old Makefile-based Kokkos to kokkos-legacy * add Kokkos 3.x packages Co-authored-by: Jeremiah J Wilke <jjwilke@kokkos-dev-2.sandia.gov>
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
# Copyright 2013-2020 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 import *
|
|
|
|
|
|
class KokkosKernelsLegacy(MakefilePackage):
|
|
"""Kokkos C++ Performance Portability Programming EcoSystem: Math Kernels -
|
|
Provides BLAS, Sparse BLAS and Graph Kernels."""
|
|
|
|
homepage = "https://github.com/kokkos/kokkos-kernels"
|
|
url = "https://github.com/kokkos/kokkos-kernels/archive/2.7.00.tar.gz"
|
|
|
|
version('2.7.00', sha256='adf4af44eadbdfbeb9ec69dd5fae4e2852bd1fbe4a69213efd199e49f4098254')
|
|
version('2.6.00', sha256='14ebf806f66b9ca73949a478b8d959be7fa1165a640935760a724d7cc0a66335')
|
|
version('2.5.00', sha256='2c2289da3a41dafd97726e90507debafbb9f5e49ca5b0f5c8d1e044a5796f000')
|
|
|
|
# make sure kokkos kernels version matches kokkos
|
|
depends_on('kokkos-legacy@2.5.00', when='@2.5.00')
|
|
depends_on('kokkos-legacy@2.6.00', when='@2.6.00')
|
|
depends_on('kokkos-legacy@2.7.00', when='@2.7.00')
|
|
|
|
patch('makefile.patch')
|
|
|
|
def edit(self, spec, prefix):
|
|
makefile = FileFilter("src/Makefile")
|
|
makefile.filter('CXX = .*', 'CXX = ' + env['CXX'])
|
|
|
|
def build(self, spec, prefix):
|
|
with working_dir('build', create=True):
|
|
makefile_path = '%s%s' % (self.stage.source_path, '/src/Makefile')
|
|
copy(makefile_path, 'Makefile')
|
|
make_args = [
|
|
'KOKKOSKERNELS_INSTALL_PATH=%s' % prefix,
|
|
'KOKKOSKERNELS_PATH=%s' % self.stage.source_path,
|
|
'KOKKOS_PATH=%s' % spec['kokkos'].prefix
|
|
]
|
|
|
|
make('build', *make_args)
|
|
|
|
def install(self, spec, prefix):
|
|
with working_dir('build', create=False):
|
|
make_args = [
|
|
'KOKKOSKERNELS_INSTALL_PATH=%s' % prefix,
|
|
'KOKKOSKERNELS_PATH=%s' % self.stage.source_path,
|
|
'KOKKOS_PATH=%s' % spec['kokkos'].prefix
|
|
]
|
|
make('install', *make_args)
|