[JAX] Fix test path for fp8 grouped gemm ag (#2262)

Fix test path so that it gets triggered Signed-off-by: Kshitij Lakhani <klakhani@nvidia.com>

[JAX] Fix test path for fp8 grouped gemm ag (#2262)
Fix test path so that it gets triggered Signed-off-by: Kshitij Lakhani <klakhani@nvidia.com>
5ec0f33b · Kshitij Lakhani · GitHub · a3b749b1 · 5ec0f33b
Unverified Commit 5ec0f33b authored Oct 13, 2025 by Kshitij Lakhani Committed by GitHub Oct 13, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

qa/L1_jax_distributed_unittest/test.sh qa/L1_jax_distributed_unittest/test.sh +1 -1

No files found.
--- a/qa/L1_jax_distributed_unittest/test.sh
+++ b/qa/L1_jax_distributed_unittest/test.sh
@@ -9,4 +9,4 @@ set -xe
 mkdir -p "$XML_LOG_DIR"

 NVTE_JAX_UNITTEST_LEVEL="L1" python3 -m pytest -c $TE_PATH/tests/jax/pytest.ini -v --junitxml=$XML_LOG_DIR/pytest.xml $TE_PATH/tests/jax/test_distributed_*
-SCRIPT_NAME=test_multi_process_distributed_grouped_gemm.py bash $TE_PATH/tests/jax/multi_process_launch.sh
+SCRIPT_NAME=$TE_PATH/tests/jax/test_multi_process_distributed_grouped_gemm.py bash $TE_PATH/tests/jax/multi_process_launch.sh