Package Details: python-jaxlib-cuda 0.6.0-1

Git Clone URL: https://aur.archlinux.org/python-jaxlib-cuda.git (read-only, click to copy)
Package Base: python-jaxlib-cuda
Description: XLA library for JAX
Upstream URL: https://github.com/jax-ml/jax/
Keywords: deep-learning google jax maching-learning xla
Licenses: Apache-2.0
Groups: jax
Conflicts: python-jaxlib
Provides: python-jaxlib
Submitter: daskol
Maintainer: daskol
Last Packager: daskol
Votes: 9
Popularity: 0.013474
First Submitted: 2023-02-12 23:18 (UTC)
Last Updated: 2025-04-20 20:54 (UTC)

Latest Comments

« First ‹ Previous 1 2 3 4 5

daskol commented on 2024-04-04 00:01 (UTC)

@bakamotokatas Thank you for the hint. Downgrading to bazel v6 solves the issue. There is no a bazel6 package in AUR thus it is vendored directly.

bakamotokatas commented on 2024-03-27 10:55 (UTC) (edited on 2024-03-27 11:38 (UTC) by bakamotokatas)

Hello,

ERROR: Traceback (most recent call last):
    File "/var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/build_bazel_rules_apple/apple/internal/rule_support.bzl", line 221, column 36, in <toplevel>
        deps_cfg = apple_common.multi_arch_split,
Error: 'apple_common' value has no field or method 'multi_arch_split'
ERROR: error loading package '@@com_github_grpc_grpc//': at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/com_github_grpc_grpc/bazel/grpc_build_system.bzl:28:6: at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/build_bazel_rules_apple/apple/ios.bzl:33:5: at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/build_bazel_rules_apple/apple/internal/ios_rules.bzl:71:5: initialization of module 'apple/internal/rule_support.bzl' failed
ERROR: /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/tsl/tsl/BUILD:460:11: error loading package '@@com_github_grpc_grpc//': at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/com_github_grpc_grpc/bazel/grpc_build_system.bzl:28:6: at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/build_bazel_rules_apple/apple/ios.bzl:33:5: at /var/tmp/pamac-build-bakamotokatas/python-jaxlib-cuda/src/bazel/ec1b63115889792c942ec63928e6e90a/external/build_bazel_rules_apple/apple/internal/ios_rules.bzl:71:5: initialization of module 'apple/internal/rule_support.bzl' failed and referenced by '@@tsl//tsl:grpc++'
ERROR: Analysis of target '//jaxlib/tools:build_wheel' failed; build aborted: Analysis failed
INFO: Elapsed time: 64.652s, Critical Path: 0.06s
INFO: 1 process: 1 internal.
ERROR: Build did NOT complete successfully
FAILED: 
ERROR: Build failed. Not running target
==> ERROR: A failure occurred in build().
    Aborting...

I am getting this error

edit: apple_common.multi_arch_split seems to have been deleted with bazel 7.

h3ss commented on 2023-11-09 15:23 (UTC)

Thanks @daskol! Confirmed fixed :)

daskol commented on 2023-11-09 13:00 (UTC)

@h3ss Fixed. Sorry for delay.

h3ss commented on 2023-10-27 17:09 (UTC)

@daskol, when you get a chance, could you please apply the same fix to this package as you did to python-jaxlib? Specifically moving environment variable assignments into the build function. Thanks!

daskol commented on 2023-06-12 19:22 (UTC) (edited on 2023-06-12 19:25 (UTC) by daskol)

@actionless @petronny @mergen I am trying to fix the error. The issue is that Arch moved to gcc-13 but nvcc supports only gcc-12 as a host compiler. It is a bit strange since this kind of issue is solved for TensorFlow and there is no such an issue for example openxla/xla at all.

actionless commented on 2023-05-10 10:48 (UTC)

i'm having this error:

/usr/include/bits/mathcalls.h(62): error: identifier "_Float32" is undefined extern _Float32 cosf32 (_Float32 __x) noexcept (true); extern _Float32 __cosf32 (_Float32 __x) noexcept (true);

mb it doesn't like latest gcc

petronny commented on 2023-03-28 11:06 (UTC) (edited on 2023-03-28 11:06 (UTC) by petronny)

I'm getting the same error like @mergen does.

Full build log: https://github.com/arch4edu/cactus/actions/runs/4542255602/jobs/8005443180

PS. It's built on a CPU-only machine.

mergen commented on 2023-03-26 08:57 (UTC) (edited on 2023-03-26 08:57 (UTC) by mergen)

My computer has cuda 12.1.0-3 installed from the community repo and I am able to use it with pytorch or pip installed jax.

However, when I try to install this package, I get the following error:


ERROR: Skipping '@org_tensorflow//tensorflow/compiler/xla/python:enable_gpu': no such package '@local_config_cuda//cuda': Repository command failed
Could not find any cuda.h matching version '' in any subdirectory:
        ''
        'include'
        'include/cuda'
        'include/*-linux-gnu'
        'extras/CUPTI/include'
        'include/cuda/CUPTI'
        'local/cuda/extras/CUPTI/include'
of:
        '/opt/cuda/extras/CUPTI/lib64'
        '/opt/cuda/lib64'
        '/opt/cuda/nvvm/lib64'
        '/opt/intel/oneapi/compiler/latest/linux/compiler/lib/intel64_lin'
        '/opt/intel/oneapi/compiler/latest/linux/lib'
        '/opt/intel/oneapi/compiler/latest/linux/lib/x64'
        '/opt/intel/oneapi/lib/intel64'
        '/opt/intel/oneapi/mkl/latest/lib/intel64'
        '/opt/intel/oneapi/tbb/latest/lib/intel64/gcc4.8'
        '/usr'
        '/usr/lib'
        '/usr/lib/libfakeroot'
        '/usr/lib/octave/8.1.0'
        '/usr/lib/root'
        '/usr/lib32'
WARNING: Target pattern parsing failed.