Accesses the tensor reference pointing to data

Inserts the operation. operation_kind -> configuration_name -> []

Accesses an element by coordinate

Gets the value pointed to

Print one or more values with optional formatting. This function provides printf-style formatted printing capabilities. It can print values

Copies data from host to device

Convert the ratio to another type. :param dtype: The target type for conversion :type dtype: type :return: The ratio converte

Move to the next pipeline stage.

Returns the stride of the layout

Before entry of the subtile load loop

Elect one thread in the warp. The elected thread gets its predicate set to true, all others obtain false.

Gets pointer to device data

Accesses the tensor reference pointing to data

Changes the size of the logical tensor. Only allocates memory if new capacity exceeds reserved capacity. To force allocation, call reset().

Set the imaginary part of the complex number

Get the data pointer from the tensor value. Extracts the data pointer (first field) from the LLVM struct value. :param loc: Optional

Returns the size of the union in bytes. :return: The size of the union.

Unpacks an element from memory

Convert from tensor object supporting __dlpack__() to a CuTe Tensor. :param tensor_dlpack: Tensor object that supports the DLPack protocol :t

Set the real part of the complex number

NOP for TMA based mainloop

Returns a pointer to the managed object

Returns a reference to the element at the logical Coord in host memory

Create a MoE persistent tile scheduler. :param params: Scheduler parameters (from host) :param offs: Cumsum tensor o

Obtains a Coord<2> from GemmCoord

Return TmaDescriptor/TensorMap

Sets runtime fields of the Atom. Some Atoms have runtime state, for example a tcgen05 MMA Atom .. code-block:: python

Total number of elements (product of all shape dimensions).

Returns the layout object's stride vector

Clears the HostTensor allocation to size/capacity = 0

Accesses the tensor reference pointing to data

Accesses the real part of the quaternion

CUTE helper to cast SMEM pointer to unsigned