Unit Test: support ncclAvg
Этот коммит содержится в:
@@ -48,7 +48,7 @@ namespace CorrectnessTests
|
||||
ReduceCorrectnessTest,
|
||||
testing::Combine(
|
||||
// Reduction operator
|
||||
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
|
||||
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin, ncclAvg),
|
||||
// Data types
|
||||
testing::Values(ncclInt8,
|
||||
ncclUint8,
|
||||
|
||||
Ссылка в новой задаче
Block a user