diff options
| author | Daniel Bershatsky | 2023-06-21 01:58:58 +0300 |
|---|---|---|
| committer | Daniel Bershatsky | 2023-06-21 01:58:58 +0300 |
| commit | bd13abaa8bf622a1b522cb98dfe69b678e70e2cc (patch) | |
| tree | afea7d6f5bb869f2c94d22dbd11ea8db916fe8bc | |
| download | aur-bd13abaa8bf622a1b522cb98dfe69b678e70e2cc.tar.gz | |
Initialize from version 3.1.0
| -rw-r--r-- | .SRCINFO | 17 | ||||
| -rw-r--r-- | PKGBUILD | 37 |
2 files changed, 54 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..994d66e16a78 --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,17 @@ +pkgbase = cutlass + pkgdesc = CUDA Templates for Linear Algebra Subroutines + pkgver = 3.1.0 + pkgrel = 1 + url = https://github.com/NVIDIA/cutlass + arch = x86_64 + license = Apache + makedepends = cmake + makedepends = cuda + makedepends = cudann + makedepends = git + depends = cuda + depends = cudann + source = cutlass-3.1.0.tar.gz::https://github.com/NVIDIA/cutlass/archive/refs/tags/v3.1.0.tar.gz + sha256sums = 821aa2e5b709a2e5d3922dcf2c5d445b4898a6ef8bac683cfb00125eafbca029 + +pkgname = cutlass diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..af40c5419eee --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,37 @@ +# Maintainer: Daniel Bershatsky <bepshatsky@yandex.ru> +pkgname=cutlass +pkgver=3.1.0 +pkgrel=1 +pkgdesc='CUDA Templates for Linear Algebra Subroutines' +arch=('x86_64') +url='https://github.com/NVIDIA/cutlass' +license=('Apache') +groups=() +depends=('cuda' 'cudann') +makedepends=('cmake' 'cuda' 'cudann' 'git') +optdepends=() +source=("$pkgname-$pkgver.tar.gz::https://github.com/NVIDIA/$pkgname/archive/refs/tags/v$pkgver.tar.gz") +sha256sums=('821aa2e5b709a2e5d3922dcf2c5d445b4898a6ef8bac683cfb00125eafbca029') + +prepare() { + export CUTLASS_NVCC_ARCHS=7.0,7.5,8.0,8.6,8.9,9.0 + export CUDACXX=/opt/cuda/bin/nvcc +} + +build() { + cmake -S $pkgname-$pkgver -B $pkgname-$pkgver/build \ + -DCUTLASS_ENABLE_CUBLAS=ON \ + -DCUTLASS_ENABLE_CUDNN=ON \ + -DCUTLASS_ENABLE_EXAMPLES=OFF \ + -DCUTLASS_INSTALL_TESTS=OFF \ + -DCUTLASS_UNITY_BUILD_ENABLED=ON + cmake --build $pkgname-$pkgver/build -j 48 +} + +package() { + cmake --install $srcdir/$pkgname-$pkgver/build --prefix $pkgdir/usr + install -Dm644 \ + $srcdir/$pkgname-$pkgver/LICENSE.txt \ + $pkgdir/usr/share/licenses/$pkgname/LICENSE + rm -rfv $pkgdir/usr/test +} |