"""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"]