RCCL should use hipPointerAttribute_t.type (#872)

Этот коммит содержится в:
Bertan Dogancay
2023-09-05 09:44:12 -06:00
коммит произвёл GitHub
родитель 2baca3a55a
Коммит 6230b5f6b3
-4
Просмотреть файл
@@ -14,11 +14,7 @@ static ncclResult_t CudaPtrCheck(const void* pointer, struct ncclComm* comm, con
WARN("%s : %s %p is not a valid pointer", opname, ptrname, pointer);
return ncclInvalidArgument;
}
#if CUDART_VERSION >= 10000
if (attr.type == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
#else
if (attr.memoryType == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
#endif
WARN("%s : %s allocated on device %d mismatchs with NCCL device %d", opname, ptrname, attr.device, comm->cudaDev);
return ncclInvalidArgument;
}