gtest: dynamically generate tests based on test machine's GPU count (#467)

* gtest: dynamically generate tests based on test machine's GPU count

* Adjust test element size and bfloat16 threshold for up to 16 GPUs
Bu işleme şunda yer alıyor:
Wenkai Du
2021-11-16 10:28:26 -08:00
işlemeyi yapan: GitHub
ebeveyn a6dba6b9dd
işleme 03a830293c
14 değiştirilmiş dosya ile 30 ekleme ve 15 silme
+1 -1
Dosyayı Görüntüle
@@ -812,7 +812,7 @@ dropback:
case ncclUint64: isMatch &= (outputU8[j] == expectedU8[j]); break;
case ncclFloat32: isMatch &= (fabs(outputF4[j] - expectedF4[j]) < 1e-5); break;
case ncclFloat64: isMatch &= (fabs(outputF8[j] - expectedF8[j]) < 1e-12); break;
case ncclBfloat16: isMatch &= (fabs((float)outputB2[j] - (float)expectedB2[j]) < 5e-2); break;
case ncclBfloat16: isMatch &= (fabs((float)outputB2[j] - (float)expectedB2[j]) < 9e-2); break;
default:
fprintf(stderr, "[ERROR] Unsupported datatype\n");
exit(0);