o n h@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) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrrl/var/www/html/construction_image-detection-poc/venv/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|)NEXETRITON__PATH 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_targetcCrB)z,Returns a unique identifier for this backendrCrrrrhash0rEzBaseBackend.hashoptionscCrB)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 rC)rrHrrr parse_options5szBaseBackend.parse_optionsstagescCrB)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. rC)rrJrHrrr add_stages=s zBaseBackend.add_stagescCrB)zK Load additional MLIR dialects into the provided `context` rC)rcontextrrr load_dialectsIzBaseBackend.load_dialectscCrB)z\ Return a map of interface modules to their device-specific implementations rCrFrrrget_module_mapPrNzBaseBackend.get_module_mapcCs,t|tsJg}d|vr|ddgg7}|S)NDztt.divisibility) isinstancer)descretrrr parse_attrWs zBaseBackend.parse_attrcKsP|dkr|ddkr|ddrdS|dkr&|ddkr&|ddr&dSdS) zX Return a string unique to each possible specialization of the argument rrQralignFrPtensorr )r0data_ptr)argtykwargsrrrget_arg_specialization_s $z"BaseBackend.get_arg_specialization)r rrr r staticmethodrrA classmethodrrrGdictobjectrIrKrMrrrOrUr\rrrrrs*  r) metaclass)r)r7r4r'abcrr dataclassesrtypingrrtypesrr`r rrrrrs