1 родитель
03e5baa67d
Коммит
ba18b77b07
1 file changed
+1
-1
@@ -628,7 +628,7 @@ ncclResult_t ncclTopoGetXmlFromGpu(struct ncclXmlNode* pciNode, uint32_t rocmDev
|
||||
if (index == -1) {
|
||||
int cudaMajor, cudaMinor;
|
||||
cudaDeviceProp devProp;
|
||||
CUDACHECK(cudaGetDeviceProperties(&devProp, dev));
|
||||
CUDACHECK(cudaGetDeviceProperties(&devProp, 0));
|
||||
cudaMajor = devProp.major; cudaMinor = devProp.minor;
|
||||
NCCLCHECK(xmlSetAttrInt(gpuNode, "sm", cudaMajor*10+cudaMinor));
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user