introduce support for ncclFloat16/half in UT

Tá an tiomantas seo le fáil i:
Edgar Gabriel
2022-08-23 19:08:38 +00:00
tuismitheoir 3c1b1ec8c8
tiomantas f6e00dec13
D'athraigh 8 comhad le 19 breiseanna agus 9 scriosta
+1 -1
Féach ar an gComhad
@@ -47,7 +47,7 @@ namespace RcclUnitTesting
{
TestBed testBed;
// Configuration
std::vector<ncclDataType_t> const& dataTypes = {ncclInt32, ncclFloat64};
std::vector<ncclDataType_t> const& dataTypes = {ncclInt32, ncclFloat64, ncclFloat16};
std::vector<int> const numElements = {1048576, 53327, 1024};
bool const inPlace = false;
bool const useManagedMem = false;