o i @sddlmZddlmZddlmZeddZeedd dZeed dd d Z ed dZ eddZ eddZ eddZ dS)) annotations)jit)corecCs||d|S)z Computes the ceiling division of :code:`x` by :code:`div` :param x: the input number :type input: Block :param div: the divisor :param div: Block r)xdivrrg/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/triton/language/standard.pycdiv s r sigmoidcCsddt| S)Nr)rexprrrr r ssoftmaxFcCs4|t|d}t|}t|d}t|||S)Nr)rmaxr sumZfdiv)rZ ieee_roundingznumZdenrrr rs  cCst||jgS)zn Returns a contiguous flattened view of :code:`x`. :param x: the input tensor :type x: Block )rviewZnumelrrrr ravel'src CsT|||}||}||}||}t|||}|||} |||} | | fS)a Transforms indices of a row-major size_i*size_j matrix into those of one where indices are row major for each group of size_j rows. For example, for size_i = size_j = 4 and size_g = 2, it will transform [[0 , 1 , 2 , 3 ], [4 , 5 , 6 , 7 ], [8 , 9 , 10, 11], [12, 13, 14, 15]] into [[0, 2, 4 , 6 ], [1, 3, 5 , 7 ], [8, 10, 12, 14], [9, 11, 13, 15]] )rminimum) ijZsize_iZsize_jZsize_gZijZsize_gjZgroup_idZoff_iZnew_iZnew_jrrr swizzle2d2s   rcCst|d|S)a' Returns a tensor filled with the scalar value 0 for the given :code:`shape` and :code:`dtype`. :param shape: Shape of the new array, e.g., (8, 16) or (8, ) :type shape: tuple of ints :param dtype: Data-type of the new array, e.g., :code:`tl.float16` :type dtype: DType r)rfull)shapedtyperrr zerosSs rcCst|j|jS)N)rrr)inputrrr zeros_like`srN)F) __future__rZ runtime.jitrrr Z_add_math_1arg_docstrr rrrrrrrrr s&