Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDADeviceAssertionHost.h98 mutable std::mutex gpu_alloc_mutex; variable
H A DCUDADeviceAssertionHost.cpp283 const std::lock_guard<std::mutex> lock(gpu_alloc_mutex); in get_uvm_assertions_ptr_for_current_device()