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

Remove the specialized sm versions for testing. It caused release wheels to ship without correct sm versions compiled in.

Browse Source 
PiperOrigin-RevId: 494854373
This commit is contained in:
Yash Katariya 2022-12-12 16:02:11 -08:00 committed by jax authors
parent 0bdb7ec042
commit 71569e1d16
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.bazelrc
View File

@ -254,9 +254,6 @@ build:rbe_linux_cuda11.8_nvcc_py3.11 --python_path="/usr/local/bin/python3.11"
build:tensorflow_testing_rbe --project_id=tensorflow-testing
common:tensorflow_testing_rbe_linux --remote_instance_name=projects/tensorflow-testing/instances/default_instance
build:tensorflow_testing_rbe_linux --config=tensorflow_testing_rbe
# Build GPU binaries for the RBE test machines (Tesla T4s i.e. sm_75) and
# kokoro (P100 i.e sm_60)
build:tensorflow_testing_rbe_linux --repo_env=TF_CUDA_COMPUTE_CAPABILITIES=sm_60,sm_75
#############################################################################
# Load `.jax_configure.bazelrc` file written by build.py