1from ctypes import c_void_p 2 3from torch import Tensor 4 5# Defined in torch/csrc/inductor/aoti_runner/pybind.cpp 6 7# Tensor to AtenTensorHandle 8def unsafe_alloc_void_ptrs_from_tensors(tensors: list[Tensor]) -> list[c_void_p]: ... 9def unsafe_alloc_void_ptr_from_tensor(tensor: Tensor) -> c_void_p: ... 10 11# AtenTensorHandle to Tensor 12def alloc_tensors_by_stealing_from_void_ptrs( 13 handles: list[c_void_p], 14) -> list[Tensor]: ... 15def alloc_tensor_by_stealing_from_void_ptr( 16 handle: c_void_p, 17) -> Tensor: ... 18 19class AOTIModelContainerRunnerCpu: ... 20class AOTIModelContainerRunnerCuda: ... 21