Skip to content

Commit 87cadd6

Browse files
committed
cuda.core: regenerate _tensor_bridge.pyi for the new ctypedefs
1 parent 128b4f5 commit 87cadd6

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

‎cuda_core/cuda/core/_tensor_bridge.pyi‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ from typing_extensions import TypeAlias
5555
AOTITorchError: TypeAlias = int
5656
AtenTensorHandle: TypeAlias = AtenTensorOpaque
5757
_get_cuda_stream_fn_t: TypeAlias = Callable[[int, Any], AOTITorchError]
58+
_tensor_from_pyobj_fn_t: TypeAlias = Callable[[Any, AtenTensorHandle], AOTITorchError]
59+
_delete_tensor_fn_t: TypeAlias = Callable[[AtenTensorHandle], AOTITorchError]
5860

5961
class PyObject(TypedDict): ...
6062

0 commit comments

Comments
 (0)