Mojo function

trace_arg

trace_arg(name: String, shape: Index[size, element_type=element_type]) -> String

Helper to stringify the type and shape of a kernel argument for tracing.

Args:

name (String): The name of the argument.
shape (Index[size, element_type=element_type]): The shape of the argument.

Returns:

A string representation of the argument with its shape.

trace_arg(name: String, shape: Index[size, element_type=element_type], dtype: DType) -> String

Helper to stringify the type and shape of a kernel argument for tracing.

Args:

name (String): The name of the argument.
shape (Index[size, element_type=element_type]): The shape of the argument.
dtype (DType): The data type of the argument.

Returns:

A string representation of the argument with its shape and data type.

trace_arg(name: String, buf: NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive]) -> String

Helper to stringify the type and shape of a kernel argument for tracing.

Args:

name (String): The name of the argument.
buf (NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive]): The NDBuffer to trace.

Returns:

A string representation of the buffer with its shape and data type.