Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyInterpreter.h208 virtual void trace_gpu_device_synchronization(
H A DPyInterpreter.cpp120 void trace_gpu_device_synchronization( in trace_gpu_device_synchronization() function
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAFunctions.cpp139 (*interp)->trace_gpu_device_synchronization(c10::kCUDA); in device_synchronize()
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUStream.cpp291 (*interp)->trace_gpu_device_synchronization(c10::kXPU); in syncStreamsOnDevice()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp131 void trace_gpu_device_synchronization( in trace_gpu_device_synchronization() function