wg ddlZddlZddlZddlmZmZmZddlmZddl m Z edGdde Z Gd d e Z y) N)ABCMetaabstractmethodabstractclassmethod) dataclass)UnionT)frozenc6eZdZUeed<eeefed<eed<y) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rint]/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/triton/backends/compiler.pyr r sL S/Nrr ceZdZdeddfdZedefdZedefdZ e defdZ e d e de fd Ze d e d e ddfd Ze d Zy) BaseBackendtargetreturnNc8||_|j|sJyN)rsupports_target)selfrs r__init__zBaseBackend.__init__s ##F+++rbinaryctjjtjjttj }tj jd|jddtjj|ddd|g}|D]}|jdd}tjj|s7tjj|sWtj|d gtj }|tj d |j#d tj$ }|||j'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_binarys77<< 9299E JJNNWV\\^$4E:B ? GGLL=&% H  3A''#,q/Cww~~c"rww~~c':#00#{1CJL]L]^% ii(A6==QXCYacamamnG* '--"222 3\&233rctrNotImplementedError)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,444 " """c"""T"f"" " " "4 " """rr) metaclass)r,r;r8abcrrr dataclassesrtypingrrWr rrrrr\sF <<! $9"G9"r