Skip to content

GitLab

"...AutoBuildImmortalWrt.git" did not exist on "d9ab41c876b92e73b57bc54261106b500243738f"
Unverified Commit 67051eff authored by zlsh80826's avatar zlsh80826 Committed by GitHub
Browse files

[JAX] Inherit the existing XLA_FLAGS for the unit tests (#491) 



Inherit the existing XLA_FLAGS
Signed-off-by: default avatarReese Wang <rewang@nvidia.com>
parent 7a025c6b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
qa/L0_jax_unittest/test.sh
View file @ 67051eff
...@@ -13,6 +13,6 @@ pip install -r $TE_PATH/examples/jax/encoder/requirements.txt ...@@ -13,6 +13,6 @@ pip install -r $TE_PATH/examples/jax/encoder/requirements.txt
pytest -Wignore -v $TE_PATH/examples/jax/mnist pytest -Wignore -v $TE_PATH/examples/jax/mnist
# Make encoder tests to have run-to-run deterministic to have the stable CI results # Make encoder tests to have run-to-run deterministic to have the stable CI results
export XLA_FLAGS="--xla_gpu_deterministic_ops" export XLA_FLAGS="${XLA_FLAGS} --xla_gpu_deterministic_ops"
pytest -Wignore -v $TE_PATH/examples/jax/encoder --ignore=$TE_PATH/examples/jax/encoder/test_multiprocessing_encoder.py pytest -Wignore -v $TE_PATH/examples/jax/encoder --ignore=$TE_PATH/examples/jax/encoder/test_multiprocessing_encoder.py
pytest -Wignore -v $TE_PATH/examples/jax/encoder/test_multiprocessing_encoder.py pytest -Wignore -v $TE_PATH/examples/jax/encoder/test_multiprocessing_encoder.py
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment