Skip to content

Commit

Permalink
Remove outdated TG llama tests from CI (old codebase)
Browse files Browse the repository at this point in the history
  • Loading branch information
johanna-rock-tt committed Jan 23, 2025
1 parent a2ac861 commit 0230b84
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 15 deletions.
9 changes: 1 addition & 8 deletions tests/scripts/tg/run_tg_frequent_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,14 +36,7 @@ run_tg_llama3_tests() {

run_tg_tests() {

if [[ "$1" == "llama3-70b-old" ]]; then
echo "LOG_METAL: running llama3_70b (old) run_tg_frequent_tests"
pytest -n auto models/demos/tg/llama3_70b/tests/test_llama_mlp_galaxy.py --timeout=300 ; fail+=$?
pytest -n auto models/demos/tg/llama3_70b/tests/test_llama_attention_galaxy.py --timeout=480 ; fail+=$?
pytest -n auto models/demos/tg/llama3_70b/tests/test_llama_decoder_galaxy.py --timeout=600 ; fail+=$?
pytest -n auto models/demos/tg/llama3_70b/tests/test_llama_model_galaxy_ci.py --timeout=800 ; fail+=$?

elif [[ "$1" == "llama3" ]]; then
if [[ "$1" == "llama3" ]]; then
echo "LOG_METAL: running Llama3 run_tg_frequent_tests"
run_tg_llama3_tests

Expand Down
1 change: 0 additions & 1 deletion tests/scripts/tg/run_tg_model_perf_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ run_tg_llm_tests() {

echo "LOG_METAL: Running run_t3000_llama2_70b_tests"
pytest -n auto models/demos/t3000/llama2_70b/tests/test_llama_perf_decode.py -m "model_perf_t3000" --timeout=600 ; fail+=$?
pytest -n auto models/demos/t3000/llama2_70b/tests/test_llama_perf_decode.py -m "model_perf_tg" --timeout=600 ; fail+=$?

# Merge all the generated reports
env python3 models/perf/merge_perf_results.py; fail+=$?
Expand Down
9 changes: 3 additions & 6 deletions tests/scripts/tg/run_tg_nightly_tests.sh
Original file line number Diff line number Diff line change
@@ -1,21 +1,18 @@
#!/bin/bash

run_tg_llama3_70b_tests() {
run_tg_tests() {
# Record the start time
fail=0
start_time=$(date +%s)

echo "LOG_METAL: Running run_tg_llama3_70b_tests"
echo "LOG_METAL: Running run_tg_tests"

pytest -n auto tests/nightly/tg/ccl --timeout=180 ; fail+=$?

# Falcon40B prefill 60 layer end to end with 10 loops; we need 8x8 grid size
pytest tests/nightly/tg/models/demos/tg/llama3_70b ; fail+=$?

# Record the end time
end_time=$(date +%s)
duration=$((end_time - start_time))
echo "LOG_METAL: run_tg_llama3_70b_tests $duration seconds to complete"
echo "LOG_METAL: run_tg_tests $duration seconds to complete"
if [[ $fail -ne 0 ]]; then
exit 1
fi
Expand Down

0 comments on commit 0230b84

Please sign in to comment.