Multi stream unit test (#693)

* Adding multi-stream support to unit tests
此提交包含在:
gilbertlee-amd
2023-02-23 13:28:50 -07:00
提交者 GitHub
父節點 d601c4909c
當前提交 80ed608a9d
共有 8 個檔案被更改,包括 165 行新增41 行删除
+2 -1
查看文件
@@ -65,9 +65,10 @@ namespace RcclUnitTesting
int rankOffset; // Global rank offset for this child
int numCollectivesInGroup; // # of collectives to run per group call
bool useBlocking; // RCCL communication with blocking or non-blocking option
int numStreamsPerGroup; // # of different streams allowed per group call
std::vector<ncclComm_t> comms; // RCCL communicators for each rank
std::vector<int> deviceIds; // Device IDs for each rank
std::vector<hipStream_t> streams; // Streams for executing collectives
std::vector<std::vector<hipStream_t>> streams; // Streams for executing collectives
std::vector<std::vector<CollectiveArgs>> collArgs; // Info for each collective for each rank
// Constructor