Multi stream unit test (#693)

* Adding multi-stream support to unit tests

[ROCm/rccl commit: 80ed608a9d]
This commit is contained in:
gilbertlee-amd
2023-02-23 13:28:50 -07:00
committed by GitHub
orang tua 8aaf6f8e0d
melakukan 0da1d6a6cd
8 mengubah file dengan 165 tambahan dan 41 penghapusan
@@ -16,6 +16,7 @@ namespace RcclUnitTesting
ncclDataType_t const dataType,
size_t const numInputElements,
size_t const numOutputElements,
int const streamIdx,
OptionalColArgs const &optionalColArgs)
{
// Free scalar based on previous scalarMode
@@ -35,6 +36,7 @@ namespace RcclUnitTesting
this->dataType = dataType;
this->numInputElements = numInputElements;
this->numOutputElements = numOutputElements;
this->streamIdx = streamIdx;
this->options = optionalColArgs;
if (this->options.scalarMode != -1)