gtest: extend testing up to 8 GPUs

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