a jh @sddlmZmZddlmZddlmZddlmZm Z ddl m Z eddGdd d e Z Gd d d eZGd d d edZdS))ABCMetaabstractmethod) dataclass)Enum)DictUnion) ModuleTypeT)frozenc@s.eZdZUeed<eeefed<eed<dS) GPUTargetbackendarchZ warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrrF/var/www/auris/lib/python3.9/site-packages/triton/backends/compiler.pyr s r c@seZdZdZdZdZdS)Languagez1The input language being compiled by the backend.rN)r rr__doc__ZTRITONZGLUONrrrrrsrc@seZdZeddddZeeedddZeedd d Z ee e d d d Z ee e ddddZ eddZeeeefdddZeddZeddZdS) BaseBackendN)targetreturncCs||_||sJdSN)rsupports_target)selfrrrr__init__szBaseBackend.__init__rcCstdSrNotImplementedErrorrrrrrszBaseBackend.supports_target)rcCstdS)z,Returns a unique identifier for this backendNr rrrrhash"szBaseBackend.hash)optionsrcCstdS)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 Nr )rr$rrr parse_options'szBaseBackend.parse_options)stagesr$rcCstdS)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. Nr )rr&r$rrr add_stages/s zBaseBackend.add_stagescCstdS)zK Load additional MLIR dialects into the provided `context` Nr )rcontextrrr load_dialects;szBaseBackend.load_dialectscCstdS)z\ Return a map of interface modules to their device-specific implementations Nr r"rrrget_module_mapBszBaseBackend.get_module_mapcCs,t|tsJg}d|vr(|ddgg7}|S)NDztt.divisibility) isinstancer)descretrrr parse_attrIs zBaseBackend.parse_attrcKsP|dkr$|ddkr$|ddr$dS|dkrL|ddkrL|ddrLdSdS) zX Return a string unique to each possible specialization of the argument rr,rZalignFr+Ztensor)getZdata_ptr)argtykwargsrrrget_arg_specializationQs  $z"BaseBackend.get_arg_specialization)r rrr r staticmethodrrrr#dictobjectr%r'r)rrr*r0r6rrrrrs$   r) metaclassN)abcrrZ dataclassesrenumrtypingrrtypesrr9r rrrrrrs