* splitting CI tests in running SP first and MP second
* set device before hipStreamSynchronize on tests
Этот коммит содержится в:
akolliasAMD
2023-03-21 08:48:39 -06:00
коммит произвёл GitHub
родитель b02fd04165
Коммит 9a0d4a07a6
2 изменённых файлов: 3 добавлений и 1 удалений
+1
Просмотреть файл
@@ -657,6 +657,7 @@ namespace RcclUnitTesting
for (int localRank : localRanksToExecute)
{
if (this->verbose) INFO("Starting synchronization for rank %d\n", localRank);
CHECK_HIP(hipSetDevice(this->deviceIds[localRank]));
for (int i = 0; i < this->numStreamsPerGroup; i++)
CHECK_HIP(hipStreamSynchronize(this->streams[localRank][i]));
}