set TE_PATH

Signed-off-by: Phuong Nguyen <[email protected]>
NVIDIA · Dec 5, 2024 · 2a923f2 · 2a923f2
1 parent 7f3efcc
commit 2a923f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/qa/L0_jax_distributed_unittest/test.sh b/qa/L0_jax_distributed_unittest/test.sh
@@ -4,6 +4,8 @@
 
 set -xe
 
+: ${TE_PATH:=/opt/transformerengine}
+
 pip install -r $TE_PATH/examples/jax/encoder/requirements.txt
 
 # Make encoder tests to have run-to-run deterministic to have the stable CI results