Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dutils_cuda.h49 aoti_torch_create_cuda_stream_guard(stream, device_index, &ptr)); in AOTICudaStreamGuard()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_cuda.cpp31 AOTITorchError aoti_torch_create_cuda_stream_guard( in aoti_torch_create_cuda_stream_guard() function
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/c/
H A Dshim.h603 AOTI_TORCH_EXPORT AOTITorchError aoti_torch_create_cuda_stream_guard(