çg ddlZddlZddlZddlmZmZmZddlmZddl m Z edGdde Z Gd d e Z dS) N)ABCMetaabstractmethodabstractclassmethod) dataclass)UnionT)frozenc>eZdZUeed<eeefed<eed<dS) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rintT/var/www/html/ai-engine/env/lib/python3.11/site-packages/triton/backends/compiler.pyr r s8LLL S/NNNNNrr ceZdZdeddfdZedefdZedefdZ e defdZ e d e de fd Ze d e d e ddfd Ze d ZdS) BaseBackendtargetreturnNcB||_||sJdSN)rsupports_target)selfrs r__init__zBaseBackend.__init__s) ##F+++++++rbinaryctjtjttj}tjd|ddtj|ddd|g}|D]}| dd}tj |rtj |rstj |d gtj }|Otjd |d tj }|||dfcSt)d|)NTRITON__PATH third_partycudabin rz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagsz Cannot find )ospathjoindirname__file__pardirenvirongetuppersplitexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)r base_dirpathspr'resultversions r_path_to_binaryzBaseBackend._path_to_binarysC7<< 9 929EE JNN:V\\^^:::B ? ? GLL=&% H H  3 3A''#,,q/Cw~~c"" 3rw~~c':': 3#0#{1CJL]^^^% i(A6==QXCYCYacamnnnG* '--"2"222222&22333rctrNotImplementedError)rs rrzBaseBackend.supports_target*s!!rct)z,Returns a unique identifier for this backendrH)rs rhashzBaseBackend.hash.s "!roptionsct)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 rH)rrLs r parse_optionszBaseBackend.parse_options3s "!rstagesct)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. rH)rrOrLs r add_stageszBaseBackend.add_stages;s "!rct)zK Load additional MLIR dialects into the provided `context` rH)rcontexts r load_dialectszBaseBackend.load_dialectsGs "!r)rrrr r staticmethodrrFrrrrKdictobjectrNrQrTrrrrrs*,y,T,,,,4444\4 " """""c"""^""T"f"""^" " " "4 " " "^ """^"""rr) metaclass)r,r;r8abcrrr dataclassesrtypingrrWr rrrrr\s <<<<<<<<<<!!!!!! $9"9"9"9"9"G9"9"9"9"9"9"r