mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
1aca76fc13
This change enables testing the wheels produced by the build rules in the presubmit using one `bazel test` command only. There are three options for running the tests: 1) `build_jaxlib=true`: the tests depend on JAX targets. 2) `build_jaxlib=false`: the tests depend on the wheel files located in the `dist` folder. 3) `build_jaxlib=wheel`: the tests depend on the py_import targets. PiperOrigin-RevId: 735765819
23 lines
259 B
Plaintext
23 lines
259 B
Plaintext
#
|
|
# test deps
|
|
#
|
|
-r test-requirements.txt
|
|
-r gpu-test-requirements.txt
|
|
|
|
#
|
|
# build deps
|
|
#
|
|
numpy~=2.0.0; python_version<="3.12"
|
|
numpy~=2.1.0; python_version>="3.13"
|
|
|
|
#
|
|
# runtime deps
|
|
#
|
|
scipy>=1.13.1
|
|
|
|
ml_dtypes>=0.4.0
|
|
opt_einsum
|
|
zstandard
|
|
etils[epath]
|
|
setuptools
|