o bZh@sddlZddlZddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z eddGdd d eZGd d d ed ZdS) N)ABCMetaabstractmethod) dataclass)DictUnion) ModuleTypeT)frozenc@s.eZdZUeed<eeefed<eed<dS) GPUTargetbackendarchZ warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrrG/var/www/auris/lib/python3.10/site-packages/triton/backends/compiler.pyr s  r c@seZdZdeddfddZedefddZee defd d Z e defd d Z e d e de fddZe de d e ddfddZe ddZe deeeffddZeddZeddZdS) BaseBackendtargetreturnNcCs||_||s JdSN)rsupports_target)selfrrrr__init__szBaseBackend.__init__binarycCs|td7}tjtjttj}tj d| ddtj|ddd|g}|D]5}tj |ratj |rat j|dgt jd }|duratjd |d tjd }|dura||d fSq,td|)NEXEZTRITON_Z_PATHZ third_partycudabinz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagsz Cannot find ) sysconfigget_config_varospathjoindirname__file__pardirenvirongetupperexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)rbase_dirpathsr&resultversionrrr_path_to_binaryszBaseBackend._path_to_binarycCtrNotImplementedError)rrrrr+zBaseBackend.supports_targetcCr>)z,Returns a unique identifier for this backendr?rrrrhash0rAzBaseBackend.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?)rrDrrr parse_options5szBaseBackend.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?)rrFrDrrr add_stages=s zBaseBackend.add_stagescCr>)zK Load additional MLIR dialects into the provided `context` r?)rcontextrrr load_dialectsIzBaseBackend.load_dialectscCr>)z\ Return a map of interface modules to their device-specific implementations r?rBrrrget_module_mapPrJzBaseBackend.get_module_mapcCs,t|tsJg}d|vr|ddgg7}|S)NDztt.divisibility) isinstancer)Zdescretrrr parse_attrWs zBaseBackend.parse_attrcKsP|dkr|ddkr|ddrdS|dkr&|ddkr&|ddr&dSdS) zX Return a string unique to each possible specialization of the argument rrMralignFrLZtensorr)r,Zdata_ptr)argtykwargsrrrget_arg_specialization_s $z"BaseBackend.get_arg_specialization)r r rr r staticmethodrr= classmethodrrrCdictobjectrErGrIrrrKrPrUrrrrrs*  r) metaclass)r%r3r0r#abcrr dataclassesrtypingrrtypesrrYr rrrrrs