8h ddlmZmZddlmZddlmZddlmZm Z ddl m Z edGdd e Z Gd d eZGd d eZdS))ABCMetaabstractmethod) dataclass)Enum)DictUnion) ModuleTypeT)frozenc>eZdZUeed<eeefed<eed<dS) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rint`/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/triton/backends/compiler.pyr r s8LLL S/NNNNNrr ceZdZdZdZdZdS)Languagez1The input language being compiled by the backend.rN)rrr__doc__TRITONGLUONrrrrrs;; F EEErrceZdZdeddfdZeedefdZedefdZ ede de fdZ ed e de ddfd Z ed Zedeeeffd Zed ZedZdS) BaseBackendtargetreturnNcB||_||sJdSN)r!supports_target)selfr!s r__init__zBaseBackend.__init__s) ##F+++++++rctr$NotImplementedError)r!s rr%zBaseBackend.supports_target "!rct)z,Returns a unique identifier for this backendr)r&s rhashzBaseBackend.hash"r+roptionsct)z Converts an `options` dictionary into an arbitrary object and returns it. This function may contain target-specific heuristics and check the legality of the provided options r))r&r/s r parse_optionszBaseBackend.parse_options's "!rstagesct)a Populates `stages` dictionary with entries of the form: ir_name [str] => Function[(src: str, metadata: dict) -> str|bytes] The value of each entry may populate a `metadata` dictionary. Stages will be run sequentially (in inseriton order) and can communicate using `metadata`. All stages are expected to return a `str` object, except for the last stage which returns a `bytes` object for execution by the launcher. r))r&r2r/s r add_stageszBaseBackend.add_stages/s "!rct)zK Load additional MLIR dialects into the provided `context` r))r&contexts r load_dialectszBaseBackend.load_dialects; "!rct)z\ Return a map of interface modules to their device-specific implementations r)r-s rget_module_mapzBaseBackend.get_module_mapBr8rcPt|tsJg}d|vr|ddggz }|S)NDztt.divisibility) isinstancer)descrets r parse_attrzBaseBackend.parse_attrIs>$$$$$$ $;; &+, ,C rc |dkr!|dzdkr|ddrdS|dkr3|dzdkr|ddrdSdS) zX Return a string unique to each possible specialization of the argument rr=ralignFr<tensor)getdata_ptr)argtykwargss rget_arg_specializationz"BaseBackend.get_arg_specializationQsm ;;38q==VZZ-G-G=3 >>cllnnr1Q666::gu;U;U63rr)rrrr r' staticmethodrr%rr.dictobjectr1r4r7rr r:rArKrrrr r sr,y,T,,,," """^\""c"""^""T"f"""^" " " "4 " " "^ """^" "S*_ 5"""^" \\rr ) metaclassN)abcrr dataclassesrenumrtypingrrtypesr rNr rr rrrrUs''''''''!!!!!! $t CCCCCGCCCCCCr