o h @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__rintrr^/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/triton/backends/compiler.pyr s  r c@seZdZdZdZdZdS)Languagez1The input language being compiled by the backend.rN)r rr__doc__ZTRITONZGLUONrrrrrsrc@seZdZdeddfddZeedefddZedefdd Z ed e de fd d Z ed e d e ddfddZ eddZedeeeffddZeddZeddZdS) BaseBackendtargetreturnNcCs||_||s JdSN)rsupports_target)selfrrrr__init__szBaseBackend.__init__cCtrNotImplementedError)rrrrrzBaseBackend.supports_targetcCr)z,Returns a unique identifier for this backendr rrrrhash"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 )rr%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 )rr'r%rrr add_stages/s zBaseBackend.add_stagescCr)zK Load additional MLIR dialects into the provided `context` r )rcontextrrr load_dialects;zBaseBackend.load_dialectscCr)z\ Return a map of interface modules to their device-specific implementations r r#rrrget_module_mapBr+zBaseBackend.get_module_mapcCs,t|tsJg}d|vr|ddgg7}|S)NDztt.divisibility) isinstancer)Zdescretrrr parse_attrIs zBaseBackend.parse_attrcKsP|dkr|ddkr|ddrdS|dkr&|ddkr&|ddr&dSdS) zX Return a string unique to each possible specialization of the argument rr.ralignFr-Ztensor)getZdata_ptr)argtykwargsrrrget_arg_specializationQs $z"BaseBackend.get_arg_specialization)r rrr r staticmethodrrrr$dictobjectr&r(r*rrr,r1r8rrrrrs&  r) metaclassN)abcrr dataclassesrenumrtypingrrtypesrr;r rrrrrrs