Revision aebb5ad0cb3033188da1e4057b0bca0ca95713fd authored by Aden Grue on 16 September 2021, 21:50:33 UTC, committed by jax authors on 29 September 2021, 00:37:41 UTC
Initially this feature was implemented only for XLA GPU; this change extends support to XLA CPU as well. As for XLA GPU, CustomCall instructions for these functions can be created via the C++ or Python builder interfaces. These functions must have the documented function signature (accepting the new `XlaCustomCallStatus*` parameter). PiperOrigin-RevId: 397174680
1 parent 45370ec
File | Mode | Size |
---|---|---|
.github | ||
benchmarks | ||
build | ||
cloud_tpu_colabs | ||
design_notes | ||
docs | ||
examples | ||
images | ||
jax | ||
jaxlib | ||
tests | ||
third_party | ||
.bazelrc | -rw-r--r-- | 12.3 KB |
.bazelversion | -rw-r--r-- | 6 bytes |
.gitignore | -rw-r--r-- | 233 bytes |
.pre-commit-config.yaml | -rw-r--r-- | 563 bytes |
.readthedocs.yml | -rw-r--r-- | 528 bytes |
CHANGELOG.md | -rw-r--r-- | 39.5 KB |
CONTRIBUTING.md | -rw-r--r-- | 150 bytes |
LICENSE | -rw-r--r-- | 11.1 KB |
README.md | -rw-r--r-- | 22.0 KB |
WORKSPACE | -rw-r--r-- | 1.3 KB |
conftest.py | -rw-r--r-- | 852 bytes |
mypy.ini | -rw-r--r-- | 523 bytes |
pylintrc | -rw-r--r-- | 1.6 KB |
pytest.ini | -rw-r--r-- | 715 bytes |
setup.cfg | -rw-r--r-- | 329 bytes |
setup.py | -rw-r--r-- | 2.5 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...