25 lines
720 B
Python
25 lines
720 B
Python
|
from ctypes import c_void_p
|
||
|
|
|
||
|
|
from torch import Tensor
|
||
|
|
|
||
|
|
# Defined in torch/csrc/inductor/aoti_runner/pybind.cpp
|
||
|
|
|
||
|
|
# Tensor to AtenTensorHandle
|
||
|
|
def unsafe_alloc_void_ptrs_from_tensors(tensors: list[Tensor]) -> list[c_void_p]: ...
|
||
|
|
def unsafe_alloc_void_ptr_from_tensor(tensor: Tensor) -> c_void_p: ...
|
||
|
|
|
||
|
|
# AtenTensorHandle to Tensor
|
||
|
|
def alloc_tensors_by_stealing_from_void_ptrs(
|
||
|
|
handles: list[c_void_p],
|
||
|
|
) -> list[Tensor]: ...
|
||
|
|
def alloc_tensor_by_stealing_from_void_ptr(
|
||
|
|
handle: c_void_p,
|
||
|
|
) -> Tensor: ...
|
||
|
|
|
||
|
|
class AOTIModelContainerRunnerCpu: ...
|
||
|
|
class AOTIModelContainerRunnerCuda: ...
|
||
|
|
class AOTIModelContainerRunnerXpu: ...
|
||
|
|
|
||
|
|
# Defined in torch/csrc/inductor/aoti_package/pybind.cpp
|
||
|
|
class AOTIModelPackageLoader: ...
|