diff options
| author | Felix Erkinger | 2022-08-16 17:29:37 +0200 |
|---|---|---|
| committer | Felix Erkinger | 2022-08-16 17:29:37 +0200 |
| commit | d026e88beb691a05dcfb2007d4692b13a3f334a2 (patch) | |
| tree | 1e5f8160aad99e35d7534b6b9f025216dd11866b | |
| parent | acf9caff0faab70efeffade7e423e5085da5e0ae (diff) | |
| download | aur-d026e88beb691a05dcfb2007d4692b13a3f334a2.tar.gz | |
also provide and conflict with python-jaxlib as we package jaxlib into jax
| -rw-r--r-- | .SRCINFO | 4 | ||||
| -rw-r--r-- | PKGBUILD | 6 |
2 files changed, 6 insertions, 4 deletions
@@ -1,7 +1,7 @@ pkgbase = python-jax-rocm pkgdesc = Differentiate, compile, and transform Numpy code (with ROCM) pkgver = 0.3.16 - pkgrel = 1 + pkgrel = 2 url = https://github.com/google/jax arch = x86_64 license = Apache @@ -23,7 +23,9 @@ pkgbase = python-jax-rocm 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 @@ -4,7 +4,7 @@ # Contributor: Daniel Bershatsky <bepshatsky@yandex.ru> pkgname='python-jax-rocm' pkgver=0.3.16 -pkgrel=1 +pkgrel=2 pkgdesc='Differentiate, compile, and transform Numpy code (with ROCM)' _srcname="jax-jax-v${pkgver}" # develop-upstream@2022-08-10:17:10:MESZ @@ -44,8 +44,8 @@ sha512sums=( '04c9ece4cb782f52925b1e7ee18ccc916a65dde051554b106164d371b3b7b96037218e635c235c8f38d088066421c0c5a4a7e201ed435b3e69c128f5ac20f0ac' '36596fd586cbdac990466a53cc0683de759b2f0646ed08edf04e88e3ee4de1a7381cf1b3a4784aa9a240e2ad894d55e8728226b95071539f774bc5d9b790b5fc' ) -conflicts=('python-jax') -provides=('python-jax') +conflicts=('python-jax' 'python-jaxlib') +provides=('python-jax' 'python-jaxlib') # test # python -c "import jax; print(jax.devices(),jax.devices()[0].device_kind); x=jax.numpy.array([1.2,3.4,5.6]); y=jax.numpy.exp(x); print(y)" |