Name	Size	Mode	Actions
bundled_inputs.cpython-36.pyc	16927	0644	editdlrm
checkpoint.cpython-36.pyc	10397	0644	editdlrm
collect_env.cpython-36.pyc	13317	0644	editdlrm
cpp_extension.cpython-36.pyc	59584	0644	editdlrm
dlpack.cpython-36.pyc	1894	0644	editdlrm
file_baton.cpython-36.pyc	1824	0644	editdlrm
hooks.cpython-36.pyc	7257	0644	editdlrm
mkldnn.cpython-36.pyc	6722	0644	editdlrm
mobile_optimizer.cpython-36.pyc	4748	0644	editdlrm
model_zoo.cpython-36.pyc	231	0644	editdlrm
show_pickle.cpython-36.pyc	4828	0644	editdlrm
throughput_benchmark.cpython-36.pyc	6730	0644	editdlrm
_cpp_extension_versioner.cpython-36.pyc	1768	0644	editdlrm
_crash_handler.cpython-36.pyc	1005	0644	editdlrm
_python_dispatch.cpython-36.pyc	780	0644	editdlrm
_pytree.cpython-36.pyc	6747	0644	editdlrm
__init__.cpython-36.pyc	730	0644	editdlrm

3 �Eg��@sjdd
lm
Z

ddlZddlZddlmZ
ddlmZ
Gdd�dej�Z ej jed�
e
ejd�d d �Z dS)�)
�AnyN)
�_from_dlpack)
� _to_dlpackc
@s0eZ
dZdZdZd ZdZdZdZdZ dZ dZd S)�DLDeviceType�
������ � �N)
r)
r)
r)
r )
r )
r)
r)
r )
r)�__name__� __module__�__qualname__ZkDLCPU�kDLGPUZkDLCPUPinnedZ kDLOpenCLZ kDLVulkanZkDLMetalZkDLVPI�kDLROCMZ kDLExtDev�rr�>/usr/local/lib64/python3.6/site-packages/torch/utils/dlpack.pyr s







rz�to_dlpack(tensor) -> PyCapsule Returns a DLPack representing the tensor. Args: tensor: a tensor to be exported The DLPack shares the tensors memory. Note that each DLPack can only be consumed once. )� ext_tensor�returnc
Csbt|d
�rV|j
�}
|
dtjtjfkrLtjjdj|
d�
�
}|j |j d�
}qZ|j �}n|}t|�
S)afrom_dlpack(ext_tensor) -> Tensor Convers a tensor from a external library into a ``torch.Tensor`` by means of the ``__dlpack__`` protocol. The tensor will share the memory with the object represented in the DLPack. .. warning:: Only call from_dlpack once per capsule. Its behavior when used on the same capsule multiple times is undefined. Args: ext_tensor (object with __dlpack__ attribute or DLPack capsule): The tensor or DLPack capsule to convert. � __dlpack__r
zcuda:{}r)
�stream)�hasattrZ__dlpack_device__rrr�torchZcudaZcurrent_stream�formatrZcuda_streamr)rZdevicerZdlpackrrr�from_dlpack$s

 
r)�typingrr�enumZtorch._CrrZ to_dlpack�IntEnumr�_CZ_add_docstrZTensorrrrrr�
s