Revision f6108674d93727fb89236a51ddc947b0f9d7a17d authored by George Necula on 11 April 2020, 15:27:14 UTC, committed by George Necula on 11 April 2020, 16:30:44 UTC
It seems that JAX (and XLA) are a lot more forgiving for index-out-of-bounds, except for the case when the dimension has size 0. In this case give a similar IndexError message as numpy, instead of the current one: RuntimeError: Invalid argument: Slice size at index 0 in gather op is out of range, must be within [0, 1), got 1.: This is a bug in JAX's shape-checking rules; please report it! Issue: #2671
1 parent c4dd1cf
File | Mode | Size |
---|---|---|
jaxlib | ||
BUILD.bazel | -rw-r--r-- | 1.3 KB |
Dockerfile | -rw-r--r-- | 1.4 KB |
build.py | -rwxr-xr-x | 11.4 KB |
build_jaxlib_wheels.sh | -rwxr-xr-x | 971 bytes |
build_jaxlib_wheels_macos.sh | -rwxr-xr-x | 1.6 KB |
build_wheel_docker_entrypoint.sh | -rwxr-xr-x | 1.4 KB |
install_cuda.sh | -rw-r--r-- | 804 bytes |
install_xla_in_source_tree.sh | -rwxr-xr-x | 3.7 KB |
setup.py | -rw-r--r-- | 1.2 KB |
Computing file changes ...