Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dutils_cuda.h30 aoti_torch_create_cuda_guard(device_index, &ptr)); in AOTICudaGuard()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_cuda.cpp8 AOTITorchError aoti_torch_create_cuda_guard( in aoti_torch_create_cuda_guard() function
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/c/
H A Dshim.h589 AOTI_TORCH_EXPORT AOTITorchError aoti_torch_create_cuda_guard(