introduce support for ncclFloat16/half in UT
[ROCm/rccl commit: f6e00dec13]
Dieser Commit ist enthalten in:
1 Ursprung
1d55fe756c
Commit
b32b819151
8 Dateien geändert
+19
-9
Keine Dateien entsprechen Ihrer Suche
@@ -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;
|
||||
|
||||
In neuem Issue referenzieren
Einen Benutzer sperren