blob: 6dd52085d614231d5103b8417e27ca2c7abe8b24 (
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
|
pkgbase = python-flashinfer-rocm
pkgdesc = ROCm port of FlashInfer (CDNA3+)
pkgver = 0.5.3
pkgrel = 2
url = https://github.com/ROCm/flashinfer
arch = x86_64
license = Apache-2.0
makedepends = git
makedepends = ninja
makedepends = python-build
makedepends = python-setuptools-scm
makedepends = python-scikit-build-core
makedepends = cmake
makedepends = rocm-hip-sdk
makedepends = python-installer
depends = python-pytorch
depends = rocm-hip-runtime
optdepends = python-click
optdepends = python-tvm-ffi
provides = python-flashinfer
source = https://github.com/ROCm/flashinfer/archive/refs/tags/v0.5.3+amd.2.tar.gz
sha256sums = f83eb02af3cbec2c757a093a7b308dd814b5e6901dcc2308070a4b3172ec9f76
pkgname = python-flashinfer-rocm
|