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

Move export backwards compatibility tests out of jax2tf. Step 2.

Browse Source 
These tests are independent of TensorFlow, yet by being in the jax2tf package they end up pulling in TensorFlow as a dependency.

This is part of a larger cl/562671314 that ran into OSS build problems.

This is step 2: moves the other test data Python files.

PiperOrigin-RevId: 591934999
This commit is contained in:
George Necula 2023-12-18 10:12:52 -08:00 committed by jax authors
parent e356d76913
commit cc2a3eb564
20 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_cholesky_lapack_potrf.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_eig_lapack_geev.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_eigh_lapack_syev.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_lu_lapack_getrf.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_qr_lapack_geqrf.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_schur_lapack_gees.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_svd_lapack_gesdd.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cpu_triangular_solve_blas_trsm.py → jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cuda_eigh_cusolver_syev.py → jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cuda_qr_cusolver_geqrf.py → jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/cuda_threefry2x32.py → jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/stablehlo_dynamic_rng_bit_generator.py → jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/stablehlo_dynamic_top_k.py → jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_ApproxTopK.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_Eigh.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_Lu.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_Qr.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_Sharding.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.py
View File

0
jax/experimental/jax2tf/tests/back_compat_testdata/tpu_stablehlo_dynamic_reduce_window.py → jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.py
View File

38
jax/experimental/jax2tf/tests/back_compat_test.py
View File

@ -31,26 +31,26 @@ from jax.experimental.export import export
from jax._src.internal_test_util import export_back_compat_test_util as bctu
from jax._src.internal_test_util.export_back_compat_test_data import cpu_ducc_fft
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_cholesky_lapack_potrf
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_eig_lapack_geev
from jax.experimental.jax2tf.tests.back_compat_testdata import cuda_eigh_cusolver_syev
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_eigh_lapack_syev
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_lu_lapack_getrf
from jax.experimental.jax2tf.tests.back_compat_testdata import cuda_qr_cusolver_geqrf
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_qr_lapack_geqrf
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_schur_lapack_gees
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_svd_lapack_gesdd
from jax.experimental.jax2tf.tests.back_compat_testdata import cpu_triangular_solve_blas_trsm
from jax.experimental.jax2tf.tests.back_compat_testdata import cuda_threefry2x32
from jax._src.internal_test_util.export_back_compat_test_data import cpu_cholesky_lapack_potrf
from jax._src.internal_test_util.export_back_compat_test_data import cpu_eig_lapack_geev
from jax._src.internal_test_util.export_back_compat_test_data import cuda_eigh_cusolver_syev
from jax._src.internal_test_util.export_back_compat_test_data import cpu_eigh_lapack_syev
from jax._src.internal_test_util.export_back_compat_test_data import cpu_lu_lapack_getrf
from jax._src.internal_test_util.export_back_compat_test_data import cuda_qr_cusolver_geqrf
from jax._src.internal_test_util.export_back_compat_test_data import cpu_qr_lapack_geqrf
from jax._src.internal_test_util.export_back_compat_test_data import cpu_schur_lapack_gees
from jax._src.internal_test_util.export_back_compat_test_data import cpu_svd_lapack_gesdd
from jax._src.internal_test_util.export_back_compat_test_data import cpu_triangular_solve_blas_trsm
from jax._src.internal_test_util.export_back_compat_test_data import cuda_threefry2x32
from jax.experimental.jax2tf.tests.back_compat_testdata import tf_call_tf_function
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_Eigh
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_Lu
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_ApproxTopK
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_Qr
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_Sharding
from jax.experimental.jax2tf.tests.back_compat_testdata import tpu_stablehlo_dynamic_reduce_window
from jax.experimental.jax2tf.tests.back_compat_testdata import stablehlo_dynamic_rng_bit_generator
from jax.experimental.jax2tf.tests.back_compat_testdata import stablehlo_dynamic_top_k
from jax._src.internal_test_util.export_back_compat_test_data import tpu_Eigh
from jax._src.internal_test_util.export_back_compat_test_data import tpu_Lu
from jax._src.internal_test_util.export_back_compat_test_data import tpu_ApproxTopK
from jax._src.internal_test_util.export_back_compat_test_data import tpu_Qr
from jax._src.internal_test_util.export_back_compat_test_data import tpu_Sharding
from jax._src.internal_test_util.export_back_compat_test_data import tpu_stablehlo_dynamic_reduce_window
from jax._src.internal_test_util.export_back_compat_test_data import stablehlo_dynamic_rng_bit_generator
from jax._src.internal_test_util.export_back_compat_test_data import stablehlo_dynamic_top_k
from jax.experimental import pjit
from jax.experimental.shard_map import shard_map