* Adding multi-stream support to unit tests
Этот коммит содержится в:
gilbertlee-amd
2023-02-23 13:28:50 -07:00
коммит произвёл GitHub
родитель d601c4909c
Коммит 80ed608a9d
8 изменённых файлов: 165 добавлений и 41 удалений
+2
Просмотреть файл
@@ -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)