mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update build.py with Cloud TPU support

Browse Source 
Update build.py to have Cloud TPU support by default.
b/180656002
This commit is contained in:
Situ Yi 2021-02-19 12:57:57 -08:00 committed by GitHub
parent 15a2f509b8
commit 77905ef5fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
build/build.py
View File

@ -271,6 +271,9 @@ build:short_logs --output_filter=DONT_MATCH_ANYTHING
# Workaround for gcc 10+ warnings related to upb.
# See https://github.com/tensorflow/tensorflow/issues/39467
build:linux --copt=-Wno-stringop-truncation
# Build with Cloud TPU support.
build --define=with_tpu_support=true
"""