Ntj 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__rintV/var/www/html/CA-Chatbot/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 EEErrc2eZdZdZdeddfdZeedefdZede fdZ ede de fd Z ed e de ddfd Zed Zedee effd ZedZedZedZdS) BaseBackendTtargetreturnNcB||_||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_stages0s "!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_mapCr8rcPt|tsJg}d|vr|ddggz }|S)NDztt.divisibility) isinstancer)descrets r parse_attrzBaseBackend.parse_attrJs>$$$$$$ $;; &+, ,C rc H|dzdkr|ddrdSdSNr=ralignFr<)getargkwargss rget_int_specializationz"BaseBackend.get_int_specializationRs+ 8q==VZZ77=3rrc l|dzdkr|ddrdSdSrC)data_ptrrFrGs rget_tensor_specializationz%BaseBackend.get_tensor_specializationXs7 <<>>B ! # # 7E(B(B #3rr)rrr%supports_native_tensor_specializationr r' staticmethodrr%rr.dictobjectr1r4r7rr r:rArJrMrrrr r s,0),y,T,,,," """^\""c"""^""T"f"""^" " " "4 " " "^ """^" "S*_ 5"""^" \\ \rr ) metaclassN)abcrr dataclassesrenumrtypingrrtypesr rQr rr rrrrXs''''''''!!!!!! $t EEEEEGEEEEEEr