File size: 319 Bytes
8e9f35a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """Small host helpers shared by the architecture backends."""
from cutlass.cute.runtime import from_dlpack
def to_cute_tensor(tensor):
return from_dlpack(
tensor,
assumed_align=16,
enable_tvm_ffi=True,
).mark_layout_dynamic(leading_dim=tensor.ndim - 1)
__all__ = ["to_cute_tensor"]
|