Adding FP16 cases to unit tests(#1093)

Signed-off-by: Tim Hu <timhu102@amd.com>
Cette révision appartient à :
Tim
2024-02-26 12:08:04 -05:00
révisé par GitHub
Parent 74f9e5db64
révision 0d06b0f1de
6 fichiers modifiés avec 63 ajouts et 14 suppressions
+1 -1
Voir le fichier
@@ -12,7 +12,7 @@ namespace RcclUnitTesting
TestBed testBed;
// Configuration
std::vector<ncclDataType_t> const& dataTypes = {ncclInt32, ncclFloat64};
std::vector<ncclDataType_t> const& dataTypes = {ncclInt32, ncclFloat16, ncclFloat64};
std::vector<int> const numElements = {1048576, 53327, 1024, 0};
bool const inPlace = false;
bool const useManagedMem = false;