tilelang.language.customize =========================== .. py:module:: tilelang.language.customize .. autoapi-nested-parse:: The language interface for tl programs. Functions --------- .. autoapisummary:: tilelang.language.customize.dp4a tilelang.language.customize.clamp tilelang.language.customize.reshape tilelang.language.customize.view tilelang.language.customize.loop_break Module Contents --------------- .. py:function:: dp4a(A, B, C) Perform a 4-element dot product with accumulation (DP4A). :param A: First input buffer :type A: Buffer :param B: Second input buffer :type B: Buffer :param C: Accumulation buffer :type C: Buffer :returns: Handle to the DP4A operation :rtype: PrimExpr .. py:function:: clamp(dst, min_val, max_val) Clamps the input value dst between [min_val, max_val] :param dst: Input value to be clamped :param min_val: Minimum value :param max_val: Maximum value :returns: Value clamped to the specified range .. py:function:: reshape(src, shape) Reshapes the input buffer to the specified shape. :param src: Input buffer to be reshaped :type src: Buffer :param shape: New shape for the buffer :type shape: List[PrimExpr] :returns: A new buffer view with the specified shape :rtype: Buffer .. py:function:: view(src, shape = None, dtype = None) Return a Tensor view of the input buffer with an optional new shape and dtype. If `shape` is None the source buffer's shape is used; if `dtype` is None the source buffer's dtype is used. The returned buffer shares the same underlying data as `src` (no copy). .. py:function:: loop_break() Break out of the current loop. :returns: A call to the `tl.loop_break` intrinsic. :rtype: tir.Call