diff options
author | Todd Gamblin <tgamblin@llnl.gov> | 2022-07-30 15:19:18 -0700 |
---|---|---|
committer | Todd Gamblin <tgamblin@llnl.gov> | 2022-07-31 13:29:20 -0700 |
commit | f52f6e99dbf1131886a80112b8c79dfc414afb7c (patch) | |
tree | 05cb7d64b2395922f2f24683da49f472075be12c /var/spack/repos/builtin/packages/magma/package.py | |
parent | 549ba1ed32372c67fc57271cde3797d58b7dec6e (diff) | |
download | spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.gz spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.bz2 spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.xz spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.zip |
black: reformat entire repository with black
Diffstat (limited to 'var/spack/repos/builtin/packages/magma/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/magma/package.py | 256 |
1 files changed, 125 insertions, 131 deletions
diff --git a/var/spack/repos/builtin/packages/magma/package.py b/var/spack/repos/builtin/packages/magma/package.py index e8f901fd89..870a6d6d50 100644 --- a/var/spack/repos/builtin/packages/magma/package.py +++ b/var/spack/repos/builtin/packages/magma/package.py @@ -9,168 +9,165 @@ from spack.package import * class Magma(CMakePackage, CudaPackage, ROCmPackage): """The MAGMA project aims to develop a dense linear algebra library similar - to LAPACK but for heterogeneous/hybrid architectures, starting with - current "Multicore+GPU" systems. + to LAPACK but for heterogeneous/hybrid architectures, starting with + current "Multicore+GPU" systems. """ homepage = "https://icl.cs.utk.edu/magma/" - git = 'https://bitbucket.org/icl/magma' + git = "https://bitbucket.org/icl/magma" url = "https://icl.cs.utk.edu/projectsfiles/magma/downloads/magma-2.2.0.tar.gz" - maintainers = ['stomov', 'luszczek', 'G-Ragghianti'] + maintainers = ["stomov", "luszczek", "G-Ragghianti"] - tags = ['e4s'] + tags = ["e4s"] test_requires_compiler = True - version('master', branch='master') - version('2.6.2', sha256='75b554dab00903e2d10b972c913e50e7f88cbc62f3ae432b5a086c7e4eda0a71', preferred=True) - version('2.6.1', sha256='6cd83808c6e8bc7a44028e05112b3ab4e579bcc73202ed14733f66661127e213') - version('2.6.0', sha256='50cdd384f44f06a34469e7125f8b2ffae13c1975d373c3f1510d91be2b7638ec') - version('2.5.4', sha256='7734fb417ae0c367b418dea15096aef2e278a423e527c615aab47f0683683b67') - version('2.5.3', sha256='c602d269a9f9a3df28f6a4f593be819abb12ed3fa413bba1ff8183de721c5ef6') - version('2.5.2', sha256='065feb85558f9dd6f4cc4db36ac633a3f787827fc832d0b578a049a43a195620') - version('2.5.1', sha256='ce32c199131515336b30c92a907effe0c441ebc5c5bdb255e4b06b2508de109f') - version('2.5.0', sha256='4fd45c7e46bd9d9124253e7838bbfb9e6003c64c2c67ffcff02e6c36d2bcfa33') - version('2.4.0', sha256='4eb839b1295405fd29c8a6f5b4ed578476010bf976af46573f80d1169f1f9a4f') - version('2.3.0', sha256='010a4a057d7aa1e57b9426bffc0958f3d06913c9151463737e289e67dd9ea608') - version('2.2.0', sha256='df5d4ace417e5bf52694eae0d91490c6bde4cde1b0da98e8d400c5c3a70d83a2') - - variant('fortran', default=True, - description='Enable Fortran bindings support') - variant('shared', default=True, - description='Enable shared library') - variant('cuda', default=True, description='Build with CUDA') - - depends_on('blas') - depends_on('lapack') - depends_on('cuda@8:', when='@2.5.1: +cuda') # See PR #14471 - depends_on('hipblas', when='+rocm') - depends_on('hipsparse', when='+rocm') - - conflicts('~cuda', when='~rocm', msg='Either CUDA or HIP support must be enabled') - conflicts('+rocm', when='+cuda', msg='CUDA must be disabled to support HIP (ROCm)') - conflicts('+rocm', when='@:2.5.4', msg='HIP support starts in version 2.6.0') - conflicts('cuda_arch=none', when='+cuda', - msg='Please indicate a CUDA arch value or values') + version("master", branch="master") + version( + "2.6.2", + sha256="75b554dab00903e2d10b972c913e50e7f88cbc62f3ae432b5a086c7e4eda0a71", + preferred=True, + ) + version("2.6.1", sha256="6cd83808c6e8bc7a44028e05112b3ab4e579bcc73202ed14733f66661127e213") + version("2.6.0", sha256="50cdd384f44f06a34469e7125f8b2ffae13c1975d373c3f1510d91be2b7638ec") + version("2.5.4", sha256="7734fb417ae0c367b418dea15096aef2e278a423e527c615aab47f0683683b67") + version("2.5.3", sha256="c602d269a9f9a3df28f6a4f593be819abb12ed3fa413bba1ff8183de721c5ef6") + version("2.5.2", sha256="065feb85558f9dd6f4cc4db36ac633a3f787827fc832d0b578a049a43a195620") + version("2.5.1", sha256="ce32c199131515336b30c92a907effe0c441ebc5c5bdb255e4b06b2508de109f") + version("2.5.0", sha256="4fd45c7e46bd9d9124253e7838bbfb9e6003c64c2c67ffcff02e6c36d2bcfa33") + version("2.4.0", sha256="4eb839b1295405fd29c8a6f5b4ed578476010bf976af46573f80d1169f1f9a4f") + version("2.3.0", sha256="010a4a057d7aa1e57b9426bffc0958f3d06913c9151463737e289e67dd9ea608") + version("2.2.0", sha256="df5d4ace417e5bf52694eae0d91490c6bde4cde1b0da98e8d400c5c3a70d83a2") + + variant("fortran", default=True, description="Enable Fortran bindings support") + variant("shared", default=True, description="Enable shared library") + variant("cuda", default=True, description="Build with CUDA") + + depends_on("blas") + depends_on("lapack") + depends_on("cuda@8:", when="@2.5.1: +cuda") # See PR #14471 + depends_on("hipblas", when="+rocm") + depends_on("hipsparse", when="+rocm") + + conflicts("~cuda", when="~rocm", msg="Either CUDA or HIP support must be enabled") + conflicts("+rocm", when="+cuda", msg="CUDA must be disabled to support HIP (ROCm)") + conflicts("+rocm", when="@:2.5.4", msg="HIP support starts in version 2.6.0") + conflicts("cuda_arch=none", when="+cuda", msg="Please indicate a CUDA arch value or values") # currently not compatible with CUDA-11 # https://bitbucket.org/icl/magma/issues/22/cuda-11-changes-issue # https://bitbucket.org/icl/magma/issues/25/error-cusparsesolveanalysisinfo_t-does-not - conflicts('^cuda@11:', when='@:2.5.3') + conflicts("^cuda@11:", when="@:2.5.3") # Many cuda_arch values are not yet recognized by MAGMA's CMakeLists.txt for target in [10, 11, 12, 13, 21, 32, 52, 53, 61, 62, 72, 86]: - conflicts('cuda_arch={}'.format(target)) + conflicts("cuda_arch={}".format(target)) # Some cuda_arch values had support added recently - conflicts('cuda_arch=37', when='@:2.5') - conflicts('cuda_arch=60', when='@:2.2') - conflicts('cuda_arch=70', when='@:2.2') - conflicts('cuda_arch=75', when='@:2.5.0') - conflicts('cuda_arch=80', when='@:2.5.3') - - patch('ibm-xl.patch', when='@2.2:2.5.0%xl') - patch('ibm-xl.patch', when='@2.2:2.5.0%xl_r') - patch('magma-2.3.0-gcc-4.8.patch', when='@2.3.0%gcc@:4.8') - patch('magma-2.5.0.patch', when='@2.5.0') - patch('magma-2.5.0-cmake.patch', when='@2.5.0') - patch('cmake-W.patch', when='@2.5.0:%nvhpc') - - @run_before('cmake') + conflicts("cuda_arch=37", when="@:2.5") + conflicts("cuda_arch=60", when="@:2.2") + conflicts("cuda_arch=70", when="@:2.2") + conflicts("cuda_arch=75", when="@:2.5.0") + conflicts("cuda_arch=80", when="@:2.5.3") + + patch("ibm-xl.patch", when="@2.2:2.5.0%xl") + patch("ibm-xl.patch", when="@2.2:2.5.0%xl_r") + patch("magma-2.3.0-gcc-4.8.patch", when="@2.3.0%gcc@:4.8") + patch("magma-2.5.0.patch", when="@2.5.0") + patch("magma-2.5.0-cmake.patch", when="@2.5.0") + patch("cmake-W.patch", when="@2.5.0:%nvhpc") + + @run_before("cmake") def generate_gpu_config(self): """If not an official release, a generation step is required to build""" spec = self.spec # 2.6.2rc1 is not an official release - should_generate = ('@master' in spec) or ('@2.6.2rc1' in spec) + should_generate = ("@master" in spec) or ("@2.6.2rc1" in spec) if not should_generate: return - backend = 'cuda' if '+cuda' in spec else 'hip' + backend = "cuda" if "+cuda" in spec else "hip" - gpu_target = '' - if '+cuda' in spec: - cuda_archs = spec.variants['cuda_arch'].value - gpu_target = ' '.join('sm_{0}'.format(i) for i in cuda_archs) + gpu_target = "" + if "+cuda" in spec: + cuda_archs = spec.variants["cuda_arch"].value + gpu_target = " ".join("sm_{0}".format(i) for i in cuda_archs) else: - gpu_target = spec.variants['amdgpu_target'].value + gpu_target = spec.variants["amdgpu_target"].value - with open('make.inc', 'w') as inc: - inc.write('FORT = true\n') - inc.write('GPU_TARGET = {0}\n'.format(gpu_target)) - inc.write('BACKEND = {0}\n'.format(backend)) + with open("make.inc", "w") as inc: + inc.write("FORT = true\n") + inc.write("GPU_TARGET = {0}\n".format(gpu_target)) + inc.write("BACKEND = {0}\n".format(backend)) - make('generate') + make("generate") def cmake_args(self): spec = self.spec options = [] - options.extend([ - '-DCMAKE_INSTALL_PREFIX=%s' % self.prefix, - '-DCMAKE_INSTALL_NAME_DIR:PATH=%s/lib' % self.prefix, - '-DBLAS_LIBRARIES=%s' % spec['blas'].libs.joined(';'), - # As of MAGMA v2.3.0, CMakeLists.txt does not use the variable - # BLAS_LIBRARIES, but only LAPACK_LIBRARIES, so we need to - # explicitly add blas to LAPACK_LIBRARIES. - '-DLAPACK_LIBRARIES=%s' % - (spec['lapack'].libs + spec['blas'].libs).joined(';') - ]) - - options += ['-DBUILD_SHARED_LIBS=%s' % - ('ON' if ('+shared' in spec) else 'OFF')] - - if spec.satisfies('%cce'): - options += ['-DCUDA_NVCC_FLAGS=-allow-unsupported-compiler'] - - if '+fortran' in spec: - options.extend([ - '-DUSE_FORTRAN=yes' - ]) - if spec.satisfies('%xl') or spec.satisfies('%xl_r'): - options.extend([ - '-DCMAKE_Fortran_COMPILER=%s' % self.compiler.f77 - ]) - - if spec.satisfies('%cce'): - options.append('-DCMAKE_Fortran_FLAGS=-ef') - - if spec.satisfies('^cuda'): - cuda_arch = self.spec.variants['cuda_arch'].value - if '@:2.2.0' in spec: - capabilities = ' '.join('sm{0}'.format(i) for i in cuda_arch) - options.extend(['-DGPU_TARGET=' + capabilities]) + options.extend( + [ + "-DCMAKE_INSTALL_PREFIX=%s" % self.prefix, + "-DCMAKE_INSTALL_NAME_DIR:PATH=%s/lib" % self.prefix, + "-DBLAS_LIBRARIES=%s" % spec["blas"].libs.joined(";"), + # As of MAGMA v2.3.0, CMakeLists.txt does not use the variable + # BLAS_LIBRARIES, but only LAPACK_LIBRARIES, so we need to + # explicitly add blas to LAPACK_LIBRARIES. + "-DLAPACK_LIBRARIES=%s" % (spec["lapack"].libs + spec["blas"].libs).joined(";"), + ] + ) + + options += ["-DBUILD_SHARED_LIBS=%s" % ("ON" if ("+shared" in spec) else "OFF")] + + if spec.satisfies("%cce"): + options += ["-DCUDA_NVCC_FLAGS=-allow-unsupported-compiler"] + + if "+fortran" in spec: + options.extend(["-DUSE_FORTRAN=yes"]) + if spec.satisfies("%xl") or spec.satisfies("%xl_r"): + options.extend(["-DCMAKE_Fortran_COMPILER=%s" % self.compiler.f77]) + + if spec.satisfies("%cce"): + options.append("-DCMAKE_Fortran_FLAGS=-ef") + + if spec.satisfies("^cuda"): + cuda_arch = self.spec.variants["cuda_arch"].value + if "@:2.2.0" in spec: + capabilities = " ".join("sm{0}".format(i) for i in cuda_arch) + options.extend(["-DGPU_TARGET=" + capabilities]) else: - capabilities = ' '.join('sm_{0}'.format(i) for i in cuda_arch) - options.extend(['-DGPU_TARGET=' + capabilities]) + capabilities = " ".join("sm_{0}".format(i) for i in cuda_arch) + options.extend(["-DGPU_TARGET=" + capabilities]) - if '@2.5.0' in spec: - options.extend(['-DMAGMA_SPARSE=OFF']) - if spec.compiler.name in ['xl', 'xl_r']: - options.extend(['-DCMAKE_DISABLE_FIND_PACKAGE_OpenMP=TRUE']) + if "@2.5.0" in spec: + options.extend(["-DMAGMA_SPARSE=OFF"]) + if spec.compiler.name in ["xl", "xl_r"]: + options.extend(["-DCMAKE_DISABLE_FIND_PACKAGE_OpenMP=TRUE"]) - if '+rocm' in spec: - options.extend(['-DMAGMA_ENABLE_HIP=ON']) - options.extend(['-DCMAKE_CXX_COMPILER=hipcc']) + if "+rocm" in spec: + options.extend(["-DMAGMA_ENABLE_HIP=ON"]) + options.extend(["-DCMAKE_CXX_COMPILER=hipcc"]) # See https://github.com/ROCmSoftwarePlatform/rocFFT/issues/322 - if spec.satisfies('^cmake@3.21.0:3.21.2'): - options.extend(['-D__skip_rocmclang=ON']) + if spec.satisfies("^cmake@3.21.0:3.21.2"): + options.extend(["-D__skip_rocmclang=ON"]) else: - options.extend(['-DMAGMA_ENABLE_CUDA=ON']) + options.extend(["-DMAGMA_ENABLE_CUDA=ON"]) return options - @run_after('install') + @run_after("install") def post_install(self): - install('magmablas/atomics.cuh', self.prefix.include) - install('control/magma_threadsetting.h', self.prefix.include) - install('control/pthread_barrier.h', self.prefix.include) - install('control/magma_internal.h', self.prefix.include) + install("magmablas/atomics.cuh", self.prefix.include) + install("control/magma_threadsetting.h", self.prefix.include) + install("control/pthread_barrier.h", self.prefix.include) + install("control/magma_internal.h", self.prefix.include) - test_src_dir = 'example' + test_src_dir = "example" - @run_after('install') + @run_after("install") def cache_test_sources(self): """Copy the example source files after the package is installed to an install test subdirectory for use during `spack test run`.""" @@ -179,19 +176,16 @@ class Magma(CMakePackage, CudaPackage, ROCmPackage): def test(self): test_dir = join_path(self.test_suite.current_test_cache_dir, self.test_src_dir) with working_dir(test_dir, create=False): - pkg_config_path = '{0}/lib/pkgconfig'.format(self.prefix) + pkg_config_path = "{0}/lib/pkgconfig".format(self.prefix) with spack.util.environment.set_env(PKG_CONFIG_PATH=pkg_config_path): - make('c') - self.run_test('./example_sparse', - purpose='MAGMA smoke test - sparse solver') - self.run_test('./example_sparse_operator', - purpose='MAGMA smoke test - sparse operator') - self.run_test('./example_v1', - purpose='MAGMA smoke test - legacy v1 interface') - self.run_test('./example_v2', - purpose='MAGMA smoke test - v2 interface') - if '+fortran' in self.spec: - make('fortran') - self.run_test('./example_f', - purpose='MAGMA smoke test - Fortran interface') - make('clean') + make("c") + self.run_test("./example_sparse", purpose="MAGMA smoke test - sparse solver") + self.run_test( + "./example_sparse_operator", purpose="MAGMA smoke test - sparse operator" + ) + self.run_test("./example_v1", purpose="MAGMA smoke test - legacy v1 interface") + self.run_test("./example_v2", purpose="MAGMA smoke test - v2 interface") + if "+fortran" in self.spec: + make("fortran") + self.run_test("./example_f", purpose="MAGMA smoke test - Fortran interface") + make("clean") |