fix tiled-mm (#38774)

This commit is contained in:
Rocco Meli 2023-07-11 14:28:37 +02:00 committed by GitHub
parent d0e39a9870
commit 364884df97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -10,11 +10,12 @@ class TiledMm(CMakePackage, CudaPackage, ROCmPackage):
"""Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt,
but ported to both NVIDIA and AMD GPUs."""
maintainers = ["mtaillefumier", "simonpintarelli"]
homepage = "https://github.com/eth-cscs/Tiled-MM/"
url = "https://github.com/eth-cscs/Tiled-MM/archive/refs/tags/v2.0.tar.gz"
git = "https://github.com/eth-cscs/Tiled-MM.git"
maintainers("mtaillefumier", "simonpintarelli", "RMeli")
version("master", branch="master")
version("2.2", sha256="6d0b49c9588ece744166822fd44a7bc5bec3dc666b836de8bf4bf1a7bb675aac")
version("2.0", sha256="ea554aea8c53d7c8e40044e6d478c0e8137d7e8b09d7cb9650703430d92cf32e")
@ -33,8 +34,8 @@ class TiledMm(CMakePackage, CudaPackage, ROCmPackage):
def cmake_args(self):
args = [
self.define_from_variant("BUILD_SHARED_LIBS", "shared"),
self.define_from_variant("TIELDMM_WITH_EXAMPLES", "examples"),
self.define_from_variant("TIELDMM_WITH_TESTS", "tests"),
self.define_from_variant("TILEDMM_WITH_EXAMPLES", "examples"),
self.define_from_variant("TILEDMM_WITH_TESTS", "tests"),
]
if "+rocm" in self.spec: