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.003181
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 6 Next › Last »

actionless commented on 2024-08-13 00:57 (UTC)

i've finally solved previously mentioned problem by adding this before the last line of build() func:

    sed -i \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_adv.h|/usr/include/cudnn_adv.h|g' \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_backend.h|/usr/include/cudnn_backend.h|g' \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_cnn.h|/usr/include/cudnn_cnn.h|g' \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_graph.h|/usr/include/cudnn_graph.h|g' \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_ops.h|/usr/include/cudnn_ops.h|g' \
        -e 's|/opt/cuda/targets/x86_64-linux/include/cudnn_version.h|/usr/include/cudnn_version.h|g' \
        bazel-jax-jaxlib-v0.4.31/external/local_config_cuda/cuda/BUILD

actionless commented on 2024-08-11 04:02 (UTC)

i'm getting this error during the build:

ERROR: /home/lie/.cache/pikaur/build/python-jaxlib-cuda/src/bazel/905847f791db658de2b823a610f64a15/external/local_config_cuda/cuda/BUIL
D:4241:8: Executing genrule @local_config_cuda//cuda:cudnn-include failed: (Exit 1): bash failed: error executing command (from target
@local_config_cuda//cuda:cudnn-include)
  (cd /home/lie/.cache/pikaur/build/python-jaxlib-cuda/src/bazel/905847f791db658de2b823a610f64a15/execroot/__main__ && \
  exec env - \
    CUDA_TOOLKIT_PATH=/opt/cuda \
    GCC_HOST_COMPILER_PATH=/usr/bin/gcc-13 \
    JAXLIB_RELEASE=0.4.30 \
    NCCL_INSTALL_PATH=/usr \
    PATH=/home/lie/bin:/usr/sbin:/sbin:/usr/condabin:/usr/local/sbin:/usr/local/bin:/usr/bin:/opt/cuda/bin:/opt/cuda/nsight_compute:/op
t/cuda/nsight_systems/bin:/opt/intel/mediasdk/bin:/usr/lib/jvm/default/bin:/usr/bin/site_perl:/usr/bin/vendor_perl:/usr/bin/core_perl \
    TF_CUDA_COMPUTE_CAPABILITIES=sm_50,sm_60,sm_70,sm_80,compute_90 \
    TF_CUDA_PATHS=/opt/cuda,/usr/lib,/usr \
  /bin/bash -c 'source external/bazel_tools/tools/genrule/genrule-setup.sh; cp -f "/opt/cuda/targets/x86_64-linux/include/cudnn.h" "baz
el-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn.h" && cp -f "/opt/cuda/targets/x86_64-linux/include/cudnn_adv.h"
"bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_adv.h" && cp -f "/opt/cuda/targets/x86_64-linux/include/cudnn
_backend.h" "bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_backend.h" && cp -f "/opt/cuda/targets/x86_64-lin
ux/include/cudnn_cnn.h" "bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_cnn.h" && cp -f "/opt/cuda/targets/x8
6_64-linux/include/cudnn_graph.h" "bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_graph.h" && cp -f "/opt/cud
a/targets/x86_64-linux/include/cudnn_ops.h" "bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_ops.h" && cp -f "
/opt/cuda/targets/x86_64-linux/include/cudnn_version.h" "bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cudnn/include/cudnn_versi
on.h" ')
# Configuration: 1f82cdfaad48bddc42a9a9db6afc8d7da3bd013e73745c6926c6e665ef1885b7
# Execution platform: @local_execution_config_platform//:platform
cp: cannot stat '/opt/cuda/targets/x86_64-linux/include/cudnn_adv.h': No such file or directory

for some reason it's looking for that lib inside cuda's install dir (/opt/cuda) instead of the correct path:

$ locate cudnn_adv.h
/usr/include/cudnn_adv.h

$ pacman -Qo /usr/include/cudnn_adv.h
/usr/include/cudnn_adv.h is owned by cudnn 9.2.1.18-1

intelfx commented on 2024-05-12 20:48 (UTC)

Does this really need to download an outdated bazel blob and install a doubly outdated gcc (that has since been dropped from the Arch repositories and needs to be built from source)?

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.

« First ‹ Previous 1 2 3 4 5 6 Next › Last »