AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding UnitTest timing summary (UT_SHOW_TIMING) (#692)

Browse Source
This commit is contained in:
gilbertlee-amd
2023-02-22 08:57:13 -07:00
committed by GitHub
parent d119c0886e
commit f63d3b1978
4 changed files with 38 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/common/EnvVars.cpp
+3 -1
View File
@@ -60,6 +60,7 @@ namespace RcclUnitTesting
verbose = GetEnvVar("UT_VERBOSE" , 0);
printValues = GetEnvVar("UT_PRINT_VALUES", 0);
maxRanksPerGpu = GetEnvVar("UT_MAX_RANKS_PER_GPU", 1);
showTiming = GetEnvVar("UT_SHOW_TIMING", 1);
// Limit number of supported reduction operators to just ncclSum if only allReduce is built
#ifdef BUILD_ALLREDUCE_ONLY
@@ -168,7 +169,8 @@ namespace RcclUnitTesting
std::make_pair("UT_REDOPS" , "List of reduction ops to test"),
std::make_pair("UT_DATATYPES" , "List of datatypes to test"),
std::make_pair("UT_MAX_RANKS_PER_GPU", "Maximum number of ranks using the same GPU"),
std::make_pair("UT_PRINT_VALUES" , "Print array values (# of values to print, < 0 for all)")
std::make_pair("UT_PRINT_VALUES" , "Print array values (# of values to print, < 0 for all)"),
std::make_pair("UT_SHOW_TIMING" , "Show timing table")
};
printf("================================================================================\n");