extending the unit-tests for multi-rank support

2022-03-15 18:53:52 -04:00
parent 0336ffdf70
commit a87d61db2b
6 changed files with 90 additions and 36 deletions
@@ -24,6 +24,7 @@ namespace RcclUnitTesting
    int  processMask; // Filter single/multi process            [UT_PROCESS_MASK]
    bool verbose;     // Show verbose TestBed output for debug  [UT_VERBOSE]
    int  printValues; // Print out input/output/expected arrays [UT_PRINT_VALUES]
+    int  maxRanksPerGpu; // Number of ranks using the same GPU  [UT_MAX_RANKS_PER_GPU]

    // Constructor that parses and collects environment variables
    EnvVars();