diff options
| author | Daniel Bershatsky | 2025-03-22 02:06:22 +0300 |
|---|---|---|
| committer | Daniel Bershatsky | 2025-03-22 02:06:22 +0300 |
| commit | 5a1cbf74ad7b7fbc0aa4488e35e6567ee5e37b2f (patch) | |
| tree | 527547af959bb05b614e4657c3b77b1093eb8fb0 /PKGBUILD | |
| parent | d3cf52e440b87a0ceccd86973e21a89e98717c54 (diff) | |
| download | aur-python-jaxlib.tar.gz | |
upgpkg: 0.5.3-1
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 24 |
1 files changed, 18 insertions, 6 deletions
@@ -1,7 +1,7 @@ # Maintainer: Daniel Bershatsky <bepshatsky@yandex.ru> pkgname=python-jaxlib -pkgver=0.5.2 +pkgver=0.5.3 pkgrel=1 pkgdesc='XLA library for JAX' arch=('x86_64') @@ -9,22 +9,34 @@ url='https://github.com/jax-ml/jax/' license=('Apache-2.0') groups=('jax') depends=('python-absl' - 'python-flatbuffers' 'python-ml-dtypes>=0.4.0' 'python-numpy' 'python-scipy') makedepends=('clang18' 'python-build' 'python-installer' 'python-setuptools' 'python-wheel') -source=("jax-${pkgver}.tar.gz::$url/archive/refs/tags/jax-v${pkgver}.tar.gz") -sha256sums=('8e9de1e012dd65fc4a9eec8af4aa2bf6782767130a5d8e1c1e342b7d658280fe') +_bazel_ver=7.4.1 +source=("jax-${pkgver}.tar.gz::$url/archive/refs/tags/jax-v${pkgver}.tar.gz" + "bazel-${_bazel_ver}-linux-x86_64::https://github.com/bazelbuild/bazel/releases/download/${_bazel_ver}/bazel-${_bazel_ver}-linux-x86_64") +sha256sums=('1094581a30ec069965f4e3e67d60262570cc3dd016adc62073bc24347b14270c' + 'c97f02133adce63f0c28678ac1f21d65fa8255c80429b588aeeba8a1fac6202b') + +prepare() { + ln -sf $(readlink bazel-${_bazel_ver}-linux-x86_64) $srcdir/jax-jax-v${pkgver}/build + chmod +x $srcdir/bazel-${_bazel_ver}-linux-x86_64 +} build() { + # Override default version. + export JAXLIB_RELEASE=$pkgver + cd $srcdir/jax-jax-v$pkgver - JAXLIB_RELEASE=$pkgver python build/build.py build \ + python build/build.py build \ --wheels=jaxlib \ + --bazel_path="$srcdir/bazel-${_bazel_ver}-linux-x86_64" \ --bazel_startup_options="--output_user_root=$srcdir/bazel"\ --bazel_options='--action_env=JAXLIB_RELEASE' \ --use_clang --clang_path='/usr/lib/llvm18/bin/clang' \ - --target_cpu_features=release + --target_cpu_features=release \ + --verbose } package() { |