Added Unit test for nccl send recv (#506)

Added Send Receive test that tests through all pairs
This commit is contained in:
akolliasAMD
2022-03-02 15:50:16 -05:00
committed by GitHub
orang tua 29ad0f5fbe
melakukan ff54e79799
8 mengubah file dengan 135 tambahan dan 9 penghapusan
+3 -2
Melihat File
@@ -175,7 +175,7 @@ namespace RcclUnitTesting
case ncclCollScatter: ss << "ncclScatter"; break;
case ncclCollAllToAll: ss << "ncclAllToAll"; break;
case ncclCollSend: ss << "ncclSend"; break;
case ncclCollRecv: ss << "ncclRevv"; break;
case ncclCollRecv: ss << "ncclRecv"; break;
default: ss << "[Unknown]"; break;
}
@@ -277,6 +277,7 @@ namespace RcclUnitTesting
return (funcType == ncclCollBroadcast ||
funcType == ncclCollReduce ||
funcType == ncclCollGather ||
funcType == ncclCollScatter);
funcType == ncclCollScatter ||
funcType == ncclCollSend); // this is incorrect but it works because in Send root is not root it is the peer
}
}