Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDADeviceAssertionHost.cpp88 void uvm_deleter(DeviceAssertionsData* uvm_assertions_ptr) { in uvm_deleter() argument
90 if (uvm_assertions_ptr) { in uvm_deleter()
91 C10_CUDA_IGNORE_ERROR(cudaFree(uvm_assertions_ptr)); in uvm_deleter()
294 DeviceAssertionsData* uvm_assertions_ptr = nullptr; in get_uvm_assertions_ptr_for_current_device() local
297 cudaMallocManaged(&uvm_assertions_ptr, sizeof(DeviceAssertionsData))); in get_uvm_assertions_ptr_for_current_device()
300 uvm_assertions_ptr, in get_uvm_assertions_ptr_for_current_device()
308 uvm_assertions_ptr, in get_uvm_assertions_ptr_for_current_device()
318 *uvm_assertions_ptr = DeviceAssertionsData(); in get_uvm_assertions_ptr_for_current_device()
322 uvm_assertions.at(device_num).reset(uvm_assertions_ptr); in get_uvm_assertions_ptr_for_current_device()
324 return uvm_assertions_ptr; in get_uvm_assertions_ptr_for_current_device()