Add support for A100 GPU and related platforms.
Add support for CUDA 11.
Add support for send/receive operations (beta).
このコミットが含まれているのは:
Sylvain Jeaugey
2020-05-12 14:40:18 -07:00
コミット 5949d96f36
54個のファイルの変更2051行の追加1254行の削除
+1 -1
ファイルの表示
@@ -590,7 +590,7 @@ ncclResult_t ncclTopoGetXmlFromGpu(struct ncclXmlNode* pciNode, nvmlDevice_t nvm
NCCLCHECK(xmlGetSub(pciNode, "nvlink", &nvlNode));
if (nvlNode == NULL) {
// NVML NVLink detection
int maxNvLinks = (sm < 60) ? 0 : (sm < 70) ? 4 : 6;
int maxNvLinks = (sm < 60) ? 0 : (sm < 70) ? 4 : (sm < 80) ? 6 : 12;
if (maxNvLinks > 0 && nvmlDev == NULL) {
WARN("No NVML device handle. Skipping nvlink detection.\n");