Add bfloat16 all reduce unit test

This commit is contained in:
Wenkai Du
2019-11-16 21:37:20 -08:00
والد 5e109ed400
کامیت bdac0256a5
3فایلهای تغییر یافته به همراه14 افزوده شده و 1 حذف شده
+2 -1
مشاهده پرونده
@@ -50,7 +50,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(1024, 1048576),
// Number of devices