blob: e9139fd461130497ac9693f8836896a6994aea06 (
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
33
34
35
36
|
pkgbase = python-jax-rocm
pkgdesc = Differentiate, compile, and transform Numpy code (with ROCM)
pkgver = 0.3.16
pkgrel = 2
url = https://github.com/google/jax
arch = x86_64
license = Apache
makedepends = bazel
makedepends = git
makedepends = miopen
makedepends = python-pip
makedepends = python-wheel
makedepends = rccl
makedepends = rocm-hip-sdk
depends = absl-py
depends = miopen
depends = python-etils
depends = python-numpy
depends = python-opt_einsum
depends = python-scipy
depends = python-six
depends = python
depends = rccl
depends = rocm-hip-runtime
provides = python-jax
provides = python-jaxlib
conflicts = python-jax
conflicts = python-jaxlib
source = python-jax-rocm-0.3.16.tar.gz::https://github.com/google/jax/archive/refs/tags/jax-v0.3.16.tar.gz
source = tensorflow-upstream-343a9e91963de6dd83e0f7470a641dca365d821f.tar.gz::https://github.com/ROCmSoftwarePlatform/tensorflow-upstream/archive/343a9e91963de6dd83e0f7470a641dca365d821f.tar.gz
source = fix-rocblas-include.patch
sha512sums = de2b16466009cfa56c46d44bbf58d2f8f293d6dbb6a0fdbc1591cd92fcf0eacbd119bd4d9c6ea4099a0443514a58e11e7c8fd8e94b73fa121d0f0a45a6849a53
sha512sums = 04c9ece4cb782f52925b1e7ee18ccc916a65dde051554b106164d371b3b7b96037218e635c235c8f38d088066421c0c5a4a7e201ed435b3e69c128f5ac20f0ac
sha512sums = 36596fd586cbdac990466a53cc0683de759b2f0646ed08edf04e88e3ee4de1a7381cf1b3a4784aa9a240e2ad894d55e8728226b95071539f774bc5d9b790b5fc
pkgname = python-jax-rocm
|