^j BUddlmZddlZddlZddlmZddlmZddlm Z dZ de d<d Z d e d <e e fZ d e d <dZde d<dZde d<dZeej%j&dZedGddZdd ddZddZ d dZy)!) annotationsN) dataclass)Path)LiteralcanaryzLiteral['canary'] MODE_CANARYloadzLiteral['load'] MODE_LOADz)tuple[Literal['canary'], Literal['load']] VALID_MODESokz Literal['ok'] STATUS_OKz not-foundzLiteral['not-found']STATUS_NOT_FOUNDz4cuda.pathfinder._dynamic_libs.dynamic_lib_subprocessT)frozenc"eZdZUded<ded<y)DynamicLibSubprocessPayloadLiteral['ok', 'not-found']status str | Noneabs_pathN)__name__ __module__ __qualname____annotations__|/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/cuda/pathfinder/_dynamic_libs/subprocess_protocol.pyrrs &&rr)errorcD||d}|||d<tj|S)Nrrr)jsondumps)rrrpayloads r%format_dynamic_lib_subprocess_payloadr$s- -3!IG   ::g rc4tjdt||gS)Nz-m)sys executableDYNAMIC_LIB_SUBPROCESS_MODULE)modelibnames r$build_dynamic_lib_subprocess_commandr+)s NND"?w OOrc |jDcgc]}|js|}}|st|d| tj|d}t |tst|d|d||jd}|jd}|tk(r4t |tst|d|d|tt|S|tk(r&|t|d|d|ttdSt|d|d|cc}w#tj $rt|d|d|ddwxYw) Nz produced no stdout payload for z" emitted invalid JSON payload for z: z emitted unexpected payload for rrr ) splitlinesstrip RuntimeErrorr!loadsJSONDecodeError isinstancedictgetr strrr)stdoutr* error_labellinelinesr#rrs r$parse_dynamic_lib_subprocess_payloadr;-s%//1 BdTZZ\T BE B k]*J7+VWWs**U2Y' gt $k]*J7+UWX_Wbcdd [[ "F{{:&H (C(+.NwkY[\c[fgh h*)hOO !!  +.NwkY[\c[fgh h*2BTRR +&FwkQST[S^_ ``' C   sk]*LWKWYZ_`bZcYfghnrrssDDD##+E)rrrrrzdict[str, str] | Nonereturnr6)r)r6r*r6r<z list[str])r7r6r*r6r8r6r<r) __future__rr!r& dataclassesrpathlibrtypingrrrr r r rr(__file__resolveparentsDYNAMIC_LIB_SUBPROCESS_CWDrr$r+r;rrrrEs# !!) )# ?#:Ey9Q 6Q =)4&4 V!(^335==a@ $$(  &   !   Pa aa a ! ar