Support bfloat16 on rest of the unit tests

This commit is contained in:
Wenkai Du
2019-11-18 14:17:29 -08:00
والد bdac0256a5
کامیت 4ca05c1297
8فایلهای تغییر یافته به همراه18 افزوده شده و 6 حذف شده
+2 -1
مشاهده پرونده
@@ -110,7 +110,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
// Number of devices