* splitting CI tests in running SP first and MP second
* set device before hipStreamSynchronize on tests

[ROCm/rccl commit: 9a0d4a07a6]
Αυτή η υποβολή περιλαμβάνεται σε:
akolliasAMD
2023-03-21 08:48:39 -06:00
υποβλήθηκε από GitHub
γονέας 5d9c3b0277
υποβολή 9daf0bc3d1
2 αρχεία άλλαξαν με 3 προσθήκες και 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]));
}