Unverified Commit 50be0299 authored by jberchtold-nvidia's avatar jberchtold-nvidia Committed by GitHub
Browse files

[JAX] Enable TE/JAX test timings in CI (#2475)



enable jax test timings in qa scripts
Signed-off-by: default avatarJeremy Berchtold <jberchtold@nvidia.com>
parent 61822061
...@@ -16,6 +16,8 @@ function test_fail() { ...@@ -16,6 +16,8 @@ function test_fail() {
RET=0 RET=0
FAILED_CASES="" FAILED_CASES=""
export NVTE_JAX_TEST_TIMING=1
: ${TE_PATH:=/opt/transformerengine} : ${TE_PATH:=/opt/transformerengine}
: ${XML_LOG_DIR:=/logs} : ${XML_LOG_DIR:=/logs}
mkdir -p "$XML_LOG_DIR" mkdir -p "$XML_LOG_DIR"
......
...@@ -18,6 +18,8 @@ function test_fail() { ...@@ -18,6 +18,8 @@ function test_fail() {
RET=0 RET=0
FAILED_CASES="" FAILED_CASES=""
export NVTE_JAX_TEST_TIMING=1
pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk" pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk"
pip3 install pytest==8.2.1 || error_exit "Failed to install pytest" pip3 install pytest==8.2.1 || error_exit "Failed to install pytest"
......
...@@ -11,6 +11,8 @@ function test_fail() { ...@@ -11,6 +11,8 @@ function test_fail() {
RET=0 RET=0
FAILED_CASES="" FAILED_CASES=""
export NVTE_JAX_TEST_TIMING=1
: ${TE_PATH:=/opt/transformerengine} : ${TE_PATH:=/opt/transformerengine}
: ${XML_LOG_DIR:=/logs} : ${XML_LOG_DIR:=/logs}
mkdir -p "$XML_LOG_DIR" mkdir -p "$XML_LOG_DIR"
......
...@@ -4,6 +4,8 @@ ...@@ -4,6 +4,8 @@
set -xe set -xe
export NVTE_JAX_TEST_TIMING=1
: ${TE_PATH:=/opt/transformerengine} : ${TE_PATH:=/opt/transformerengine}
: ${XML_LOG_DIR:=/logs} : ${XML_LOG_DIR:=/logs}
mkdir -p "$XML_LOG_DIR" mkdir -p "$XML_LOG_DIR"
......
...@@ -18,6 +18,8 @@ function test_fail() { ...@@ -18,6 +18,8 @@ function test_fail() {
RET=0 RET=0
FAILED_CASES="" FAILED_CASES=""
export NVTE_JAX_TEST_TIMING=1
pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk" pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk"
pip3 install pytest==8.2.1 || error_exit "Failed to install pytest" pip3 install pytest==8.2.1 || error_exit "Failed to install pytest"
......
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