diff options
Diffstat (limited to 'src/graph/xml.cc')
-rw-r--r-- | src/graph/xml.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/xml.cc b/src/graph/xml.cc index f138d0b..2885787 100644 --- a/src/graph/xml.cc +++ b/src/graph/xml.cc @@ -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"); |