Skip to content

Commit ab6ac04

Browse files
committed
fix(scripts): rename multi-process test cases
- add an _8_proc suffix to multi-process test case IDs - print manual comparison commands when no baseline log directory is set
1 parent 260f13d commit ab6ac04

2 files changed

Lines changed: 14 additions & 10 deletions

File tree

scripts/run_models_and_profile.bash

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -587,6 +587,10 @@ else
587587
echo -e "\033[1;33m To enable comparison, set 'variables.COMPARE_LOG_DIR' in ${CONFIG_FILE}\033[0m"
588588
echo -e "\033[1;33m or export COMPARE_LOG_DIR=/path/to/baseline_logs before running.\033[0m"
589589
echo -e "\033[1;33m============================================================\033[0m"
590+
591+
echo -e "\n\033[1;33mRun comparison manually:\033[0m"
592+
echo "python3 compare_loss.py \"/path/to/baseline_logs\" \"$(realpath "$LOG_DIR")\""
593+
echo "python3 compare_tps.py \"/path/to/baseline_logs\" \"$(realpath "$LOG_DIR")\""
590594
fi
591595

592596
echo -e "\n\033[1;36m[END OF TEST] Cleaning build directory after all tests\033[0m"

scripts/test_config.json

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -903,7 +903,7 @@
903903
},
904904
"tests": [
905905
{
906-
"id": "3",
906+
"id": "3_8proc",
907907
"args": {
908908
"dtype": "float32",
909909
"nthread_per_process": 1,
@@ -913,7 +913,7 @@
913913
}
914914
},
915915
{
916-
"id": "3_bfloat16",
916+
"id": "3_bfloat16_8proc",
917917
"args": {
918918
"dtype": "bfloat16",
919919
"nthread_per_process": 1,
@@ -923,7 +923,7 @@
923923
}
924924
},
925925
{
926-
"id": "4",
926+
"id": "4_8proc",
927927
"args": {
928928
"dtype": "float32",
929929
"nthread_per_process": 1,
@@ -934,7 +934,7 @@
934934
}
935935
},
936936
{
937-
"id": "4_bfloat16",
937+
"id": "4_bfloat16_8proc",
938938
"args": {
939939
"dtype": "bfloat16",
940940
"nthread_per_process": 1,
@@ -945,7 +945,7 @@
945945
}
946946
},
947947
{
948-
"id": "5",
948+
"id": "5_8proc",
949949
"args": {
950950
"dtype": "float32",
951951
"nthread_per_process": 1,
@@ -957,7 +957,7 @@
957957
}
958958
},
959959
{
960-
"id": "5_bfloat16",
960+
"id": "5_bfloat16_8proc",
961961
"args": {
962962
"dtype": "bfloat16",
963963
"nthread_per_process": 1,
@@ -969,7 +969,7 @@
969969
}
970970
},
971971
{
972-
"id": "6",
972+
"id": "6_8proc",
973973
"args": {
974974
"dtype": "float32",
975975
"nthread_per_process": 1,
@@ -980,7 +980,7 @@
980980
}
981981
},
982982
{
983-
"id": "6_bfloat16",
983+
"id": "6_bfloat16_8proc",
984984
"args": {
985985
"dtype": "bfloat16",
986986
"nthread_per_process": 1,
@@ -991,7 +991,7 @@
991991
}
992992
},
993993
{
994-
"id": "8",
994+
"id": "8_8proc",
995995
"args": {
996996
"dtype": "float32",
997997
"nthread_per_process": 1,
@@ -1005,7 +1005,7 @@
10051005
}
10061006
},
10071007
{
1008-
"id": "8_bfloat16",
1008+
"id": "8_bfloat16_8proc",
10091009
"args": {
10101010
"dtype": "bfloat16",
10111011
"nthread_per_process": 1,

0 commit comments

Comments
 (0)