For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo function
transpose_trivial_memcpy
def transpose_trivial_memcpy[dtype: DType, //](output: TileTensor[dtype, Storage=output.Storage, linear_idx_type=output.linear_idx_type], input: TileTensor[dtype, Storage=input.Storage, linear_idx_type=input.linear_idx_type], ctx: Optional[DeviceContext] = None)
Copies the input buffer to the output buffer as a trivial transpose.
Uses a single memcpy for small transfers and a parallel memcpy for
larger ones.
Parameters:
- dtype (
DType): Element type of theinputandoutputbuffers (inferred).
Args:
- output (
TileTensor[dtype, Storage=output.Storage, linear_idx_type=output.linear_idx_type]): The output buffer. - input (
TileTensor[dtype, Storage=input.Storage, linear_idx_type=input.linear_idx_type]): The input buffer. - ctx (
Optional[DeviceContext]): The context to execute the work on.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!