gtest: extend testing up to 8 GPUs

Этот коммит содержится в:
Wenkai Du
2020-06-27 21:30:50 +00:00
родитель 964c4c2061
Коммит 8db0aa8f4c
10 изменённых файлов: 14 добавлений и 14 удалений
+2 -2
Просмотреть файл
@@ -59,9 +59,9 @@ namespace CorrectnessTests
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
testing::Values(2520, 3026520),
// Number of devices
testing::Values(2,3,4),
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));