jax
c78ca042 - Add experimental support for building JAX CPU and GPU wheels with GCC.

Commit
1 year ago
Add experimental support for building JAX CPU and GPU wheels with GCC. The `build.py` script uses Clang compiler by default, and JAX doesn't support building with GCC officially. However, experimental GCC support is still present. Command examples: ``` python build/build.py build --wheels=jaxlib,jax-cuda-plugin --use_clang=false python build/build.py build --wheels=jaxlib,jax-cuda-plugin --use_clang=false --gcc_path=/use/bin/gcc ``` This change addresses the request in https://github.com/jax-ml/jax/issues/25488. PiperOrigin-RevId: 707930913
Parents
Loading