tvm_ffi.from_dlpack

Contents

tvm_ffi.from_dlpack#

tvm_ffi.from_dlpack(ext_tensor, *, require_alignment=0, require_contiguous=False)#

Import a foreign array that implements the DLPack producer protocol.

Parameters:

  • ext_tensor (object) – An object supporting __dlpack__ and __dlpack_device__.

  • require_alignment (int, optional) – If greater than zero, require the underlying data pointer to be aligned to this many bytes. Misaligned inputs raise ValueError.

  • require_contiguous (bool, optional) – When True, require the layout to be contiguous. Non-contiguous inputs raise ValueError.

Return type:

Tensor

Returns:

Tensor – A TVM FFI Tensor that references the same memory.

Examples

import numpy as np
x_np = np.arange(8, dtype="int32")
x = tvm_ffi.from_dlpack(x_np)
y_np = np.from_dlpack(x)
assert np.shares_memory(x_np, y_np)
Contents