U ®9%e ã@s’ddlmZddlmZddlmZedd„ƒZee d¡d d„ƒƒZee d ¡dd d „ƒƒZ ed d„ƒZ edd„ƒZ edd„ƒZ edd„ƒZ 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©)ÚxÚdivrrúW/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/triton/language/standard.pyÚcdiv s r ÚsigmoidcCsddt | ¡S)Nr)rÚexp©rrrr r sÚsoftmaxFcCs4|t |d¡}t |¡}t |d¡}t |||¡S)Nr)rÚmaxr ÚsumZfdiv)rZ ieee_roundingÚzÚnumZdenrrr rs  cCst ||jg¡S)zn Returns a contiguous flattened view of :code:`x`. :param x: the input tensor :type x: Block )rÚviewZnumelrrrr Ú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]] )rÚminimum) ÚiÚjZsize_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)rÚfull)ÚshapeÚdtyperrr ÚzerosSs rcCst|j|jƒS)N)rrr)Úinputrrr Ú zeros_like`srN)F)Ú __future__rZ runtime.jitrÚrr Z_add_math_1arg_docstrr rrrrrrrrr Ús$