introduce support for ncclFloat16/half in UT

[ROCm/rccl commit: f6e00dec13]
Dieser Commit ist enthalten in:
Edgar Gabriel committed 2022-08-24 15:28:24 +00:00
1 Ursprung 1d55fe756c
Commit b32b819151
8 Dateien geändert
+19 -9

Keine Dateien entsprechen Ihrer Suche

+1 -1
Datei anzeigen
@@ -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;