5^(hddlZddlZddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z edGdd eZGd d e Zy) N)ABCMetaabstractmethod) dataclass)DictUnion) ModuleTypeT)frozenc6eZdZUeed<eeefed<eed<y) GPUTargetbackendarch warp_sizeN)__name__ __module__ __qualname__str__annotations__rintV/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/triton/backends/compiler.pyr r sL S/Nrr ceZdZdeddfdZedefdZee defdZ e defdZ e d e de fd Ze d e d e ddfd Ze d Ze deeeffdZedZedZy) BaseBackendtargetreturnNc8||_|j|sJyN)rsupports_target)selfrs r__init__zBaseBackend.__init__s ##F+++rbinaryc|tjdz }tjj tjj t tj}tjjd|jddtjj |ddd|g}|D]}tjj|s#tjj|sCtj|dgtj }|mt!j"d |j%d t j& }|||j)d fcSt+d|)NEXETRITON__PATH third_partycudabinz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagsz Cannot find ) sysconfigget_config_varospathjoindirname__file__pardirenvirongetupperexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)r!base_dirpathsr0resultversions r_path_to_binaryzBaseBackend._path_to_binarys")**51177<< 9299E JJNNWV\\^$4E:B ? GGLL=&% H  6Dww~~d#t(<#00$ 1DZM^M^_% ii(A6==QXCYacamamnG*#W]]1%555  6\&233rctrNotImplementedError)rs rrzBaseBackend.supports_target+ "!rct)z,Returns a unique identifier for this backendrIrs rhashzBaseBackend.hash0rKroptionsct)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 rI)rrOs r parse_optionszBaseBackend.parse_options5s "!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. rI)rrRrOs r add_stageszBaseBackend.add_stages=s "!rct)zK Load additional MLIR dialects into the provided `context` rI)rcontexts r load_dialectszBaseBackend.load_dialectsI "!rct)z\ Return a map of interface modules to their device-specific implementations rIrMs rget_module_mapzBaseBackend.get_module_mapPrXrcFt|tsJg}d|vr|ddggz }|S)NDztt.divisibility) isinstancer)descrets r parse_attrzBaseBackend.parse_attrWs6$$$$ $; &+, ,C rc |dk(r|dzdk(r|jddry|dk(r)|jdzdk(r|jddryy) zX Return a string unique to each possible specialization of the argument rr]ralignFr\tensorr&)r6data_ptr)argtykwargss rget_arg_specializationz"BaseBackend.get_arg_specialization_sQ ;38q=VZZ-G >cllnr1Q66::gu;Ur)rrrr r staticmethodrrG classmethodrrrNdictobjectrQrTrWrrrZrarirrrrrs,y,T,444 " """c"""T"f"" " " "4 " """ "S*_ 5"" rr) metaclass)r/r=r:r-abcrr dataclassesrtypingrrtypesrrmr rrrrrssK '! $TGTr