Ë ¹K i§ãóˆ—UddlmZddlmZmZddlmZmZmZddl m Z e dd¬«Z de d <Gd „d «Z Gd „d «Zy)é)Ú annotations)ÚCallableÚOptional)ÚExecutorÚ as_completedÚFuture)Ú ContextVarÚasync_compile_active_modeN)Údefaultz&ContextVar[Optional[AsyncCompileMode]]Ú active_modecó—eZdZdd„Zd„Zy)Ú FutureKernelcó.—||_d|_||_y©N)Úfinalize_compileÚkernelÚfuture)Úselfrrs úc/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/triton/runtime/_async_compile.pyÚ__init__zFutureKernel.__init__ s€Ø 0ˆÔ؈Œ ؈ ócóš—|j |jS|jj«}|j|«||_|Sr)rrÚresultr)rrs rrzFutureKernel.resultsB€Ø ;‰;Ð "Ø—;‘;Ð à—‘×#Ñ#Ó%ˆØ ×јfÔ%؈Œ ؈ rN)rrrr)Ú__name__Ú __module__Ú __qualname__rr©rrrr s „óó rrcó&—eZdZdd„Zd„Zd„Zd„Zy)ÚAsyncCompileModecó.—||_g|_i|_yr)ÚexecutorÚ raw_futuresÚfuture_kernels)rr!s rrzAsyncCompileMode.__init__s€Ø ˆŒ ؈ÔØ ˆÕrcóô—|jj|«}||S|jj|«}||_|j j |«t||«}||j|<|Sr)r#Úgetr!ÚsubmitÚ_keyr"Úappendr)rÚkeyÚ compile_fnÚ finalize_fnrÚ future_kernels rr&zAsyncCompileMode.submit!st€Ø×$Ñ$×(Ñ(¨Ó-ˆØ Р؈Mà—‘×%Ñ% jÓ1ˆØˆŒ Ø ×Ñ×Ñ Ô'Ü$ [°&Ó9ˆ Ø#0ˆ×јCÑ ØÐrcón—tj« td«‚tj|«|S)Nz*Another AsyncCompileMode is already active)r r%Ú RuntimeErrorÚset)rs rÚ __enter__zAsyncCompileMode.__enter__-s,€Ü ?‰?Ó Ð (ÜÐKÓLÐ L܉˜Ô؈ rcó°—t|j«D])}|j|jj «Œ+t j d«yr)rr"r#r'rr r/)rÚexc_typeÚ exc_valueÚ tracebackrs rÚ__exit__zAsyncCompileMode.__exit__3sC€ä" 4×#3Ñ#3Ó4ò 6ˆFØ × Ñ  § ¡ Ñ ,× 3Ñ 3Õ 5ð 6䉘ÕrN)r!r)rrrrr&r0r5rrrrrs„ó!ò òó rr)Ú __future__rÚtypingrrÚconcurrent.futuresrrrÚ contextvarsr r Ú__annotations__rrrrrúr;s<ðÞ"ß%ß=Ñ=Ý"á6@ÐA\ÐfjÔ6k€ Ð 3Ók÷ñ÷"òr