o Gۂi @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) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrr[/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/triton/backends/compiler.pyr s  r c@seZdZdZdZdZdS)Languagez1The input language being compiled by the backend.rN)rrr__doc__TRITONGLUONrrrrrsrc@seZdZdZdeddfddZeedefddZede fd d Z ed e de fd d Z ede d e ddfddZeddZedee effddZeddZeddZeddZdS) BaseBackendTtargetreturnNcCs||_||s JdSN)rsupports_target)selfrrrr__init__szBaseBackend.__init__cCtrNotImplementedError)rrrrrzBaseBackend.supports_targetcCr")z,Returns a unique identifier for this backendr#r rrrhash#r%zBaseBackend.hashoptionscCr")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(rrr parse_options(szBaseBackend.parse_optionsstagescCr")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 r*r(rrr add_stages0s zBaseBackend.add_stagescCr")zK Load additional MLIR dialects into the provided `context` r#)r contextrrr load_dialects<zBaseBackend.load_dialectscCr")z\ Return a map of interface modules to their device-specific implementations r#r&rrrget_module_mapCr.zBaseBackend.get_module_mapcCs,t|tsJg}d|vr|ddgg7}|S)NDztt.divisibility) isinstancer)descretrrr parse_attrJs zBaseBackend.parse_attrcKs |ddkr|ddrdSdSNr1ralignFr0)getargkwargsrrrget_int_specializationRsz"BaseBackend.get_int_specializationcKs$|ddkr|ddrdSdSr6)data_ptrr9r:rrrget_tensor_specializationXsz%BaseBackend.get_tensor_specialization)rrr%supports_native_tensor_specializationr r! staticmethodrrrr'dictobjectr)r+r-rrr/r5r=r?rrrrrs,   r) metaclassN)abcrr dataclassesrenumrtypingrrtypesrrCr rrrrrrs