blob: fd93bb8f1ffd6fdabe58eec3aa4ca8a8828e4659 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
pkgbase = cutlass
pkgdesc = CUDA Templates for Linear Algebra Subroutines
pkgver = 4.3.5
pkgrel = 1
url = https://github.com/NVIDIA/cutlass
arch = x86_64
license = BSD-3-Clause
makedepends = cmake
makedepends = git
makedepends = ninja
makedepends = pybind11
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = cuda
depends = cudnn
source = cutlass-4.3.5.tar.gz::https://github.com/NVIDIA/cutlass/archive/refs/tags/v4.3.5.tar.gz
sha256sums = 73d8c3914a6049ff5c43b7dfb9d70f26e44dc9e10e36049db5a999b9faf6dbf0
pkgname = cutlass
pkgname = python-cutlass
pkgdesc = Python interface to CUTLASS library
arch = any
depends = python
depends = python-cuda
depends = python-pydot
depends = python-scikit-build
depends = python-treelib
depends = pybind11
optdepends = python-ml-dtypes: bfloat16 dtype support (alternative)
|