# Copyright 2013-2023 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.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. """ homepage = "https://icl.cs.utk.edu/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") tags = ["e4s"] test_requires_compiler = True version("master", branch="master") version("2.7.2", sha256="729bc1a70e518a7422fe7a3a54537a4741035a77be3349f66eac5c362576d560") version("2.7.1", sha256="d9c8711c047a38cae16efde74bee2eb3333217fd2711e1e9b8606cbbb4ae1a50") version("2.7.0", sha256="fda1cbc4607e77cacd8feb1c0f633c5826ba200a018f647f1c5436975b39fd18") version("2.6.2", sha256="75b554dab00903e2d10b972c913e50e7f88cbc62f3ae432b5a086c7e4eda0a71") 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") depends_on("python", when="@master", type="build") conflicts("~cuda", when="~rocm", msg="magma: Either CUDA or HIP support must be enabled") conflicts("+rocm", when="+cuda", msg="magma: CUDA must be disabled to support HIP (ROCm)") conflicts("+rocm", when="@:2.5.4", msg="magma: HIP support starts in version 2.6.0") conflicts( "cuda_arch=none", when="+cuda", msg="magma: 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") # 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)) # Some cuda_arch values had support added recently conflicts("cuda_arch=37", when="@:2.5", msg="magma: cuda_arch=37 needs a version > 2.5") conflicts("cuda_arch=60", when="@:2.2", msg="magma: cuda_arch=60 needs a version > 2.2") conflicts("cuda_arch=70", when="@:2.2", msg="magma: cuda_arch=70 needs a version > 2.2") conflicts("cuda_arch=75", when="@:2.5.0", msg="magma: cuda_arch=75 needs a version > 2.5.0") conflicts("cuda_arch=80", when="@:2.5.3", msg="magma: cuda_arch=80 needs a version > 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) if not should_generate: return 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) else: 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)) make("generate") def cmake_args(self): spec = self.spec define = self.define options = [ define("CMAKE_INSTALL_PREFIX", self.prefix), define("CMAKE_INSTALL_NAME_DIR", self.prefix.lib), define("BLAS_LIBRARIES", spec["blas"].libs), # 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. define("LAPACK_LIBRARIES", spec["lapack"].libs + spec["blas"].libs), self.define_from_variant("BUILD_SHARED_LIBS", "shared"), ] if spec.satisfies("%cce"): options.append(define("CUDA_NVCC_FLAGS", "-allow-unsupported-compiler")) if "+fortran" in spec: options.append(define("USE_FORTRAN", True)) if spec.satisfies("%xl") or spec.satisfies("%xl_r"): options.append(define("CMAKE_Fortran_COMPILER", self.compiler.f77)) if spec.satisfies("%cce"): options.append(define("CMAKE_Fortran_FLAGS", "-ef")) if "+cuda" in spec: cuda_arch = spec.variants["cuda_arch"].value sep = "" if "@:2.2.0" in spec else "_" capabilities = " ".join("sm{0}{1}".format(sep, i) for i in cuda_arch) options.append(define("GPU_TARGET", capabilities)) archs = ";".join("%s" % i for i in cuda_arch) options.append(define("CMAKE_CUDA_ARCHITECTURES", archs)) if "@2.5.0" in spec: options.append(define("MAGMA_SPARSE", False)) if spec.compiler.name in ["xl", "xl_r"]: options.append(define("CMAKE_DISABLE_FIND_PACKAGE_OpenMP", True)) if "+rocm" in spec: options.append(define("MAGMA_ENABLE_HIP", True)) options.append(define("CMAKE_CXX_COMPILER", spec["hip"].hipcc)) # See https://github.com/ROCmSoftwarePlatform/rocFFT/issues/322 if spec.satisfies("^cmake@3.21.0:3.21.2"): options.append(define("__skip_rocmclang", True)) else: options.append(define("MAGMA_ENABLE_CUDA", True)) return options @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) test_src_dir = "example" @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`.""" self.cache_extra_test_sources([self.test_src_dir]) 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) 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")