o 灛i @stddlZddlZddlZddlmZmZmZddlmZddl m Z eddGddde Z Gd d d ed Z dS) N)ABCMetaabstractmethodabstractclassmethod) dataclass)UnionT)frozenc@s.eZdZUeed<eeefed<eed<dS) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrrZ/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/triton/backends/compiler.pyr s  rc@seZdZdeddfddZedefddZedefd d Z e defd d Z e d e de fddZe de d e ddfddZe ddZdS) BaseBackendtargetreturnNcCs||_||s JdSN)rsupports_target)selfrrrr__init__szBaseBackend.__init__binarycCstjtjttj}tjd|ddtj|ddd|g}|D]<}| dd}tj |ratj |rat j |d gt jd }|duratjd |d tjd }|dura||dfSq%td|)NTRITON__PATH third_partycudabin rz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagsz Cannot find )ospathjoindirname__file__pardirenvirongetuppersplitexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)rbase_dirpathspr!resultversionrrr_path_to_binaryszBaseBackend._path_to_binarycCtrNotImplementedError)rrrrr*szBaseBackend.supports_targetcCrA)z,Returns a unique identifier for this backendrB)rrrrhash.szBaseBackend.hashoptionscCrA)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 rB)rrErrr parse_options3szBaseBackend.parse_optionsstagescCrA)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. rB)rrGrErrr add_stages;s zBaseBackend.add_stagescCrA)zK Load additional MLIR dialects into the provided `context` rB)rcontextrrr load_dialectsGszBaseBackend.load_dialects)r r rrr staticmethodrr@rrrrDdictobjectrFrHrJrrrrrs r) metaclass)r&r5r2abcrrr dataclassesrtypingrrMrrrrrrs