U h @stddlZddlZddlZddlmZmZmZddlmZddl m Z eddGddde Z Gd d d ed Z dS) N)ABCMetaabstractmethodabstractclassmethod) dataclass)UnionT)frozenc@s.eZdZUeed<eeefed<eed<dS) GPUTargetbackendarchZ warp_sizeN)__name__ __module__ __qualname__str__annotations__rintrrJ/var/www/html/venv/lib/python3.8/site-packages/triton/backends/compiler.pyr s rc@seZdZeddddZeedddZeedd d Z e ed d d Z e e e dddZe e e ddddZe ddZdS) BaseBackendN)targetreturncCs||_||stdSN)rsupports_targetAssertionError)selfrrrr__init__szBaseBackend.__init__)binarycCstjtjttj}tjd|ddtj|ddd|g}|D]x}| dd}tj |rJtj |rJt j |d gt jd }|dk rJtjd |d tjd }|dk rJ||dfSqJtd|dS)NZTRITON_Z_PATHZ third_partycudabin rz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagsz Cannot find )ospathjoindirname__file__pardirenvirongetuppersplitexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)rbase_dirpathsprresultversionrrr_path_to_binaryszBaseBackend._path_to_binaryrcCstdSrNotImplementedErrorr>rrrr*szBaseBackend.supports_target)rcCstdS)z,Returns a unique identifier for this backendNr?)rrrrhash.szBaseBackend.hash)optionsrcCstdS)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 Nr?)rrBrrr parse_options3szBaseBackend.parse_options)stagesrBrcCstdS)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. Nr?)rrDrBrrr add_stages;s zBaseBackend.add_stagescCstdS)zK Load additional MLIR dialects into the provided `context` Nr?)rcontextrrr load_dialectsGszBaseBackend.load_dialects)r r r rr staticmethodrr=rrrrAdictobjectrCrErGrrrrrs r) metaclass)r#r2r/abcrrr dataclassesrtypingrrJrrrrrrs