a zþd`ã@sbddlmZddlmZddlmZedƒZdZdddœdd„Zd d œd d „Z dddœd d„Z dS)é)Ú annotations)Úwraps)ÚTypeVarÚTZ__triton_builtin__)ÚfnÚreturncs0tˆƒs J‚tˆƒ‡fdd„ƒ}t|tdƒ|S)zMark a function as a builtin.cs*d|vs|ddurtdƒ‚ˆ|i|¤ŽS)NZ_builderzdDid you forget to add @triton.jit ? (`_builder` argument must be provided outside of JIT functions.))Ú ValueError)ÚargsÚkwargs©r©úY/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/triton/impl/base.pyÚwrappers ÿzbuiltin..wrapperT)ÚcallablerÚsetattrÚTRITON_BUILTIN)rrr r r Úbuiltin s   rÚbool)rcCs t|tdƒS)z-Is this a registered triton builtin function?F)Úgetattrrr r r r Ú is_builtinsrcCst|ƒS)z#A decorator for external functions.)rr r r r Úextern"srN) Ú __future__rÚ functoolsrÚtypingrrrrrrr r r r Ús