Home
last modified time | relevance | path

Searched refs:driver_api (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
DCUDASymmetricMemory.cu35 auto driver_api = c10::cuda::DriverAPI::get(); in device_has_multicast_support() local
37 C10_CUDA_DRIVER_CHECK(driver_api->cuDeviceGetAttribute_( in device_has_multicast_support()
41 return driver_api->cuMulticastCreate_ != nullptr && multicast_supported; in device_has_multicast_support()
246 auto driver_api = c10::cuda::DriverAPI::get(); in map_block() local
249 driver_api->cuMemAddressReserve_(dev_ptr, size, 0ULL, 0, 0ULL)); in map_block()
250 C10_CUDA_DRIVER_CHECK(driver_api->cuMemMap_(*dev_ptr, size, 0, handle, 0ULL)); in map_block()
257 C10_CUDA_DRIVER_CHECK(driver_api->cuMemSetAccess_(*dev_ptr, size, &desc, 1)); in map_block()
312 auto driver_api = c10::cuda::DriverAPI::get(); in ~CUDASymmetricMemory() local
314 C10_CUDA_DRIVER_CHECK(driver_api->cuMemUnmap_( in ~CUDASymmetricMemory()
316 C10_CUDA_DRIVER_CHECK(driver_api->cuMemRelease_(handles_[r])); in ~CUDASymmetricMemory()
[all …]
DCudaDMAConnectivity.cpp50 auto driver_api = c10::cuda::DriverAPI::get(); in detect() local
54 driver_api->nvmlDeviceGetHandleByPciBusId_v2_( in detect()
64 ret = driver_api->nvmlDeviceGetNvLinkRemoteDeviceType_( in detect()
78 driver_api->nvmlDeviceGetNvLinkRemotePciInfo_v2_( in detect()
/external/pytorch/c10/cuda/
DCMakeLists.txt39 driver_api.cpp