Searched refs:CUDAStreamGuardHandle (Results 1 – 3 of 3) sorted by relevance
34 CUDAStreamGuardHandle* ret_guard) { in aoti_torch_create_cuda_stream_guard()39 *ret_guard = reinterpret_cast<CUDAStreamGuardHandle>(guard); in aoti_torch_create_cuda_stream_guard()44 CUDAStreamGuardHandle guard) { in aoti_torch_delete_cuda_stream_guard()
22 reinterpret_cast<CUDAStreamGuardHandle>(ptr))); in delete_cuda_stream_guard()47 CUDAStreamGuardHandle ptr = nullptr; in AOTICudaStreamGuard()
601 using CUDAStreamGuardHandle = CUDAStreamGuardOpaque*; variable606 CUDAStreamGuardHandle* ret_guard // returns new reference610 aoti_torch_delete_cuda_stream_guard(CUDAStreamGuardHandle guard);