Revision 5b576cb03e85f2e4d8d99f314cf5d87981ef5d42 authored by Peter Hawkins on 27 June 2022, 20:56:00 UTC, committed by jax authors on 27 June 2022, 20:56:32 UTC
This change appears to be causing crashes on Mac. Original description: Change the pocketfft custom kernel in jaxlib to generate its flatbuffer descriptor in C++ instead. Surprisingly this code is actually much more readable in C++ because the flatbuffers Python API does not have a readable but less efficient API. Breaking changes to the flatbuffers Python APIs have caused breakage in JAX in the past, and we can avoid the dependency completely without much work. PiperOrigin-RevId: 457559793
1 parent 5b865ed
| File | Mode | Size |
|---|---|---|
| .github | ||
| benchmarks | ||
| build | ||
| cloud_tpu_colabs | ||
| docs | ||
| examples | ||
| images | ||
| jax | ||
| jaxlib | ||
| tests | ||
| third_party | ||
| .bazelrc | -rw-r--r-- | 14.0 KB |
| .bazelversion | -rw-r--r-- | 6 bytes |
| .gitignore | -rw-r--r-- | 291 bytes |
| .pre-commit-config.yaml | -rw-r--r-- | 563 bytes |
| .readthedocs.yml | -rw-r--r-- | 569 bytes |
| CHANGELOG.md | -rw-r--r-- | 62.6 KB |
| CITATION.bib | -rw-r--r-- | 408 bytes |
| CONTRIBUTING.md | -rw-r--r-- | 150 bytes |
| LICENSE | -rw-r--r-- | 11.1 KB |
| README.md | -rw-r--r-- | 23.3 KB |
| WORKSPACE | -rw-r--r-- | 1.7 KB |
| conftest.py | -rw-r--r-- | 856 bytes |
| mypy.ini | -rw-r--r-- | 618 bytes |
| pylintrc | -rw-r--r-- | 1.6 KB |
| pytest.ini | -rw-r--r-- | 1.4 KB |
| setup.cfg | -rw-r--r-- | 486 bytes |
| setup.py | -rw-r--r-- | 3.4 KB |
Computing file changes ...
