summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorJakub Klinkovský2023-12-10 11:45:54 +0100
committerJakub Klinkovský2023-12-10 11:45:54 +0100
commit234d860214e5f1198a7362b0eaba8478b85f8c27 (patch)
tree25cfae303d7f7204f67a9299557213ed8a822035
downloadaur-234d860214e5f1198a7362b0eaba8478b85f8c27.tar.gz
initial commit
-rw-r--r--.SRCINFO15
-rw-r--r--PKGBUILD39
2 files changed, 54 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..4a773da7c426
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,15 @@
+pkgbase = cutlass-headers
+ pkgdesc = CUDA Templates for Linear Algebra Subroutines (headers only)
+ pkgver = 3.3.0
+ pkgrel = 1
+ url = https://github.com/NVIDIA/cutlass
+ arch = x86_64
+ license = BSD-3-Clause
+ makedepends = cmake
+ makedepends = ninja
+ depends = cuda
+ conflicts = cutlass
+ source = cutlass-headers-3.3.0.tar.gz::https://github.com/NVIDIA/cutlass/archive/refs/tags/v3.3.0.tar.gz
+ sha256sums = 99876de94a04d0b9cdee81d4f29659bde1426b416926aef4d9a3a3a7869511a3
+
+pkgname = cutlass-headers
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..dccbf483633a
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,39 @@
+# Maintainer: Jakub Klinkovský <lahwaacz@archlinux.org>
+# Contributor: Daniel Bershatsky <bepshatsky@yandex.ru>
+
+pkgname=cutlass-headers
+pkgver=3.3.0
+pkgrel=1
+pkgdesc="CUDA Templates for Linear Algebra Subroutines (headers only)"
+arch=(x86_64)
+url="https://github.com/NVIDIA/cutlass"
+license=('BSD-3-Clause')
+depends=(cuda)
+makedepends=(cmake ninja)
+conflicts=(cutlass)
+source=("$pkgname-$pkgver.tar.gz::$url/archive/refs/tags/v$pkgver.tar.gz")
+sha256sums=('99876de94a04d0b9cdee81d4f29659bde1426b416926aef4d9a3a3a7869511a3')
+
+prepare() {
+ cd cutlass-$pkgver
+
+ # https://github.com/NVIDIA/cutlass/issues/1248
+ # delete last two lines (comment and bad add_library(...)
+ head -n -2 cmake/NvidiaCutlassConfig.cmake > cmake/NvidiaCutlassConfig.cmake
+}
+
+build() {
+ # this should not actually build anything, we want to install just the headers
+ cmake -S cutlass-$pkgver -B build -G Ninja \
+ -DCMAKE_BUILD_TYPE=None \
+ -DCMAKE_INSTALL_PREFIX=/usr \
+ -DCUTLASS_ENABLE_HEADERS_ONLY=ON \
+ -DCUTLASS_INSTALL_TESTS=OFF
+ cmake --build build
+}
+
+package() {
+ DESTDIR="$pkgdir" cmake --install build
+ install -Dm644 cutlass-$pkgver/LICENSE.txt "$pkgdir"/usr/share/licenses/$pkgname/LICENSE
+ rm -rfv "$pkgdir"/usr/test
+}