Issue 1:
Please:

Check for duplicate issues.

Provide a complete example of how to reproduce the bug, wrapped in triple backticks like this:

I've been trying to build jaxlib v.0.3.10 from source on a new cluster with cuda 11.3.1 and cudnn 8.2.1 but it seems

to be that the bazel builder is having issues with fetching llvm-raw as its returning an input/output error.

Operating system RHEL 8

CUDA version 11.3.1

cuDNN version 8.2.1.32

Python version 3.9.5

compiler GCC 10.3.0

$ git clone https://github.com/google/jax
$ cd jax

$ python build/build.py --enable_cuda --enable_nccl --cuda_path='/usr/local/easybuild/software/CUDA/11.3.1/' --cudnn_path='/usr/local/easybuild/software/cuDNN/8.2.1.32-CUDA-11.3.1/' --cuda_version='11.3' --cudnn_version='8.2.1' --target_cpu_features='native'

If applicable, include full error messages/tracebacks.

_ _ __ __
| | / \ \ \/ /
_ | |/ _ \ \ /
| |_| / ___ \/ \
\___/_/ \/_/\_\

Bazel binary path: ./bazel-5.1.1-linux-x86_64
Bazel version: 5.1.1
Python binary path: /usr/local/easybuild/software/Python/3.9.5-GCCcore-10.3.0/bin/python
Python version: 3.9
NumPy version: 1.21.0
MKL-DNN enabled: yes
Target CPU: x86_64
Target CPU features: native
CUDA enabled: yes
CUDA toolkit path: /usr/local/easybuild/software/CUDA/11.3.1/
CUDNN library path: /usr/local/easybuild/software/cuDNN/8.2.1.32-CUDA-11.3.1/
CUDA version: 11.3
CUDNN version: 8.2.1
NCCL enabled: yes
TPU enabled: no
ROCm enabled: no

Building XLA and installing it in the jaxlib source tree...
./bazel-5.1.1-linux-x86_64 run --verbose_failures=true --config=native_arch_posix --config=mkl_open_source_only --config=cuda :build_wheel -- --output_path=/home/oxg34/user_modules/jax_0.3.10/jax/dist --cpu=x86_64
Starting local Bazel server and connecting to it...
INFO: Options provided by the client:
Inherited 'common' options: --isatty=0 --terminal_columns=80
INFO: Reading rc options for 'run' from /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc:
Inherited 'common' options: --experimental_repo_remote_exec
INFO: Reading rc options for 'run' from /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc:
Inherited 'build' options: --apple_platform_type=macos --macos_minimum_os=10.9 --announce_rc --define open_source_build=true --spawn_strategy=standalone --enable_platform_specific_config --experimental_cc_shared_library --define=no_aws_support=true --define=no_gcp_support=true --define=no_hdfs_support=true --define=no_kafka_support=true --define=no_ignite_support=true --define=grpc_no_ares=true -c opt --config=short_logs --copt=-DMLIR_PYTHON_PACKAGE_PREFIX=jaxlib.mlir.
INFO: Reading rc options for 'run' from /home/oxg34/user_modules/jax_0.3.10/jax/.jax_configure.bazelrc:
Inherited 'build' options: --strategy=Genrule=standalone --repo_env PYTHON_BIN_PATH=/usr/local/easybuild/software/Python/3.9.5-GCCcore-10.3.0/bin/python --action_env=PYENV_ROOT --python_path=/usr/local/easybuild/software/Python/3.9.5-GCCcore-10.3.0/bin/python --action_env CUDA_TOOLKIT_PATH=/usr/local/easybuild/software/CUDA/11.3.1/ --action_env CUDNN_INSTALL_PATH=/usr/local/easybuild/software/cuDNN/8.2.1.32-CUDA-11.3.1/ --action_env TF_CUDA_PATHS=/usr/local/easybuild/software/CUDA/11.3.1/,/usr/local/easybuild/software/cuDNN/8.2.1.32-CUDA-11.3.1/ --action_env TF_CUDA_VERSION=11.3 --action_env TF_CUDNN_VERSION=8.2.1 --distinct_host_configuration=false
INFO: Found applicable config definition build:short_logs in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --output_filter=DONT_MATCH_ANYTHING
INFO: Found applicable config definition build:native_arch_posix in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --copt=-march=native --host_copt=-march=native
INFO: Found applicable config definition build:mkl_open_source_only in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --define=tensorflow_mkldnn_contraction_kernel=1
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
INFO: Found applicable config definition build:linux in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --config=posix --copt=-Wno-stringop-truncation --copt=-Wno-array-parameter
INFO: Found applicable config definition build:posix in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --copt=-fvisibility=hidden --copt=-Wno-sign-compare --cxxopt=-std=c++14 --host_cxxopt=-std=c++14
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: no such package '@llvm-raw//utils/bazel': java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: no such package '@llvm-raw//utils/bazel': java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
write (Input/output error)
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: no such package '@llvm-raw//utils/bazel': java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
write (Input/output error)
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: no such package '@llvm-raw//utils/bazel': java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
write (Input/output error)
INFO: Repository llvm-raw instantiated at:
/home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: in <toplevel>
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/tensorflow/workspace3.bzl:32:9: in workspace
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/llvm/workspace.bzl:10:20: in repo
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:128:21: in tf_http_archive
Repository rule _tf_http_archive defined at:
/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl:81:35: in <toplevel>
ERROR: An error occurred during the fetch of repository 'llvm-raw':
Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
ERROR: /home/oxg34/user_modules/jax_0.3.10/jax/WORKSPACE:28:14: fetching _tf_http_archive rule //external:llvm-raw: Traceback (most recent call last):
File "/home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/org_tensorflow/third_party/repo.bzl", line 64, column 33, in _tf_http_archive_impl
ctx.download_and_extract(
Error in download_and_extract: java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
INFO: Found applicable config definition build:cuda in file /home/oxg34/user_modules/jax_0.3.10/jax/.bazelrc: --repo_env TF_NEED_CUDA=1 --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_52,sm_60,sm_70,compute_80 --crosstool_top=@local_config_cuda//crosstool:toolchain --@local_config_cuda//:enable_cuda --define=xla_python_enable_gpu=true
write (Input/output error)
ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: no such package '@llvm-raw//utils/bazel': java.io.IOException: Error extracting /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076/e2ed3fd71e08ac50ca326c79f31247e7e4a16b7b.tar.gz to /home/oxg34/.cache/bazel/_bazel_oxg34/6b511a965134ca496153611be50e71ba/external/llvm-raw/temp11546401468023270076: Input/output error
write (Input/output error)
b''
Traceback (most recent call last):
File "/home/oxg34/user_modules/jax_0.3.10/jax/build/build.py", line 528, in <module>
main()
File "/home/oxg34/user_modules/jax_0.3.10/jax/build/build.py", line 523, in main
shell(command)
File "/home/oxg34/user_modules/jax_0.3.10/jax/build/build.py", line 53, in shell
output = subprocess.check_output(cmd)
File "/usr/local/easybuild/software/Python/3.9.5-GCCcore-10.3.0/lib/python3.9/subprocess.py", line 424, in check_output
return run(*popenargs, stdout=PIPE, timeout=timeout, check=True,
File "/usr/local/easybuild/software/Python/3.9.5-GCCcore-10.3.0/lib/python3.9/subprocess.py", line 528, in run
raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['./bazel-5.1.1-linux-x86_64', 'run', '--verbose_failures=true', '--config=native_arch_posix', '--config=mkl_open_source_only', '--config=cuda', ':build_wheel', '--', '--output_path=/home/oxg34/user_modules/jax_0.3.10/jax/dist', '--cpu=x86_64']' returned non-zero exit status 2.

Issue 2:
Implement numpy.partition and numpy.argpartition (currently in the docs but unimplemented). Among other things, this allows for efficient computation of order statistics (see selection algorithms) without having to sort the entire array:

import numpy as np
from timeit import default_timer

def order1(a, k, axis=-1):
return np.partition(a, k, axis).take(k, axis)

def order2(a, k, axis=-1):
return np.sort(a, axis).take(k, axis)

def test_order(shape, axis):
a = np.random.rand(*shape)
k = np.random.randint(shape[axis])

start = default_timer()
ans1 = order1(a, k, axis)
print('partition-based: {:g} seconds'.format(default_timer() - start))

start = default_timer()
ans2 = order2(a, k, axis)
print('sort-based: {:g} seconds'.format(default_timer() - start))

assert np.array_equal(ans1, ans2)

for shape in [(10**7,), (10**4, 10**3), (10**3, 10**2, 10**2)]:
for axis in range(len(shape)):
print()
test_order(shape, axis)

partition-based: 0.12672 seconds
sort-based: 0.878069 seconds

partition-based: 0.300224 seconds
sort-based: 0.679816 seconds

partition-based: 0.0932873 seconds
sort-based: 0.38649 seconds

partition-based: 0.16949 seconds
sort-based: 0.445229 seconds

partition-based: 0.134436 seconds
sort-based: 0.273936 seconds

partition-based: 0.130433 seconds
sort-based: 0.260393 seconds