Peter Hawkins hawkinsp
hawkinsp
/ JAX exported symbols
Last active
August 16, 2022 22:12
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| jax | |
| ShapeDtypeStruct | |
| ShapedArray | |
| _src | |
| abstract_arrays | |
| ad | |
| ad_checkpoint | |
| api_util | |
| block_until_ready | |
| check_tracer_leaks |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| diff --git a/build/build.py b/build/build.py | |
| index 72df7f4..a0cc2d9 100755 | |
| --- a/build/build.py | |
| +++ b/build/build.py | |
| @@ -162,7 +162,6 @@ build --action_env PYTHON_BIN_PATH="{python_bin_path}" | |
| build --python_path="{python_bin_path}" | |
| build --action_env TF_NEED_CUDA="{tf_need_cuda}" | |
| build --action_env CUDA_TOOLKIT_PATH="{cuda_toolkit_path}" | |
| -build --action_env CUDNN_INSTALL_PATH="{cudnn_install_path}" | |
| build --distinct_host_configuration=false |