`i ddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZddlZddlZddlZddlmZddlmZddlmZmZddlmZddlmZddlmZdd l m!Z!dd l m"Z"dd l m#Z#dd l$m%Z%dd l$m&Z&ddl m'Z'ddl m(Z(dZ)e*ej+ej,fZ-dej.krddlm/Z/e e/dZ0ne1Z0ej2dgdZ3Gdde4Z5dZ6dZ7GddZ8dZ9dZ:dIde e1d e1d!e e1fd"Z;d#e#j<d!e*fd$Z=Gd%d&Z>d'Z?d(Z@d)ejAd*e ee&e%fd+e>d!ee&e%ffd,ZBd-ejCd*e ee&e%fd.e ejDd+e>d!e%f d/ZEd0e ejFd1e*d+e>d!efd2ZGe6d3ejFd1e*d+e>d!efd4ZHe6d5ejId+e>d!e#j<fd6ZJd5e ejId+e>d!e#j<fd7ZKd8ee&e%fd9e%d+e>d!efd:ZL dJd;ejId+e>dZNd?e#j<d@e#j<d+e>d!ee%e&ffdAZOedBe&e%ee&e%fZPdCePdDe!jQdEe0d+e>d!ePf dFZRdCee&e%fdGee&e%fd+e>d!e%fdHZSdS)KN)AnyDictListOptionalSequenceTupleTypeVarUnion)ComplexWarning)runtime) CodeBlock _CodeType)_kernel)_raise_if_invalid_cast)jit) _cuda_types)_cuda_typerules)_internal_typesDataConstant)_builtin_funcs) _interfaceF))Literal)noequivsafe same_kindunsafeResult) func_namecode return_typeenable_cooperative_groupsbackendoptionsjitifyceZdZdZdZdS)_JitCompileErrorcdt||_t||_||_dSN)type error_typestrmesnode)selfer3s f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/jit/_compile.py__init__z_JitCompileError.__init__7s'q''q66 c|jjt|jddfdt |dD}||jdz|z)N end_lineno ctg|]4\}}|dzcxkrkrnnd|nd|5S)z>  )rstrip).0ilineendstarts r6 z,_JitCompileError.reraise..?sn:::4"QU1111c11111[$[[[{D{{ J J L L:::r8z )r3linenogetattrjoin enumeratesplitr0r2)r4pycoderCrDs @@r6reraisez_JitCompileError.reraise<s  diu55:::::$V\\$%7%788:::;;oodh/&8999r8N)__name__ __module__ __qualname__r7rLr8r6r,r,5s2 :::::r8r,c,dtjffd }|S)Nr3cv |g|Ri|S#t$rt$r}t||d}~wwxYwr.)r, Exception)r3argskwargsr5funcs r6new_funcz,transpile_function_wrapper..new_funcFsl ,4.t...v.. .     , , ,"1d++ + ,s 838)astAST)rVrWs` r6transpile_function_wrapperrZEs1,sw,,,,,, Or8c t|std tj|}n#t$rd}YnwxYw|dkrt d|d|jdkrtj|\}}t|dt|d z d fd |D}tj |}t|tjsJt|jd ksJ|jd|fS|td |d d t!j|}tj|\} t|z d |}tj |} fd tj|D}t|d krtd|d|d}tjd|jtj|jggdd|fS)a/Returns the tuple of ``ast.FunctionDef`` object and the source string for the given callable ``func``. ``func`` can be a ``def`` function or a ``lambda`` expression. The source is returned only for informational purposes (i.e., rendering an exception message in case of an error). z!`func` must be a callable object.Nzz/JIT needs access to the Python source code for zf but it cannot be retrieved within the Python interactive interpreter. Consider using IPython instead.zrcDg|]}|dzddS) r\r=)replace)r@rB num_indents r6rEz*_parse_function_object..psAEEE6:DLLz)2q 1 1EEEr8r=z+JIT needs access to Python source code for zX but could not be located. (hint: it is likely you passed a built-in function or method)crg|]3}t|tjr|jcxkrk.nn|4SrP) isinstancerXLambdarF)r@r3end_line start_lines r6rEz*_parse_function_object..sZ 7 7 7d4,, 7t{5555X55555555r8z4Multiple callables are found near the definition of z4, and JIT could not identify the source code for it._lambda_kernel)namerTbodydecorator_listreturns type_comment)callable ValueErrorinspect getsourcefile TypeError RuntimeErrorrMgetsourcelineslenlstriprHrXparserbModulerh linecachegetlineswalk FunctionDefrTReturn) rVfilenamelines_sourcetree full_sourcenodesr3rdr`res @@@r6_parse_function_objectrQs D>>><=== (.. 9 CEEEFFy  $ +++++49~~""""y|V## L$ L L LMM M''),X6677K /55FJCKK'H WWV__F 9[ ! !D 7 7 7 7 7chtnn 7 7 7E 5zzA~~=+/===>> > 8D ? DIj## $4d     s8 AAc&eZdZddZdeddfdZdS) GeneratedreturnNcg|_i|_d|_d|_d|_d|_d|_tjrdnd|_ d|_ dS)NF)z-DCUPY_JIT_MODEz --std=c++17z-DCUB_DISABLE_BF16_SUPPORTnvccnvrtc) codesdevice_function enable_cg include_cginclude_cg_memcpy_asyncinclude_cuda_barrierr)r is_hipr(r*)r4s r6r7zGenerated.__init__s` " 68 ',$$)!7 ")LIq 99Y_ % %DGG  F#I d $K"+Wg O O OOr8c L||f|jvr"|j||f}|td|St|\}}tj|} t di| j| j| j} d |}|j } t|jdkr'| dtt|jzz } d|j||f<t|| ||| |||| \} } | | jf|j||f<|| | | jfS)Nz$Recursive function is not supported.r^rr~)rrP)rrmrrngetclosurevarsdictglobals nonlocalsbuiltinsrHrgrsr1_transpile_functionrr)rVrrrrrresultrrcvarsconstsrg cuda_codeenvs r6rrsP h9444*D(+;< >CDD D *$//LD&  "4 ( (E  G GEM GU_ G G GF*%%J 9D 9 $%%)) c#i788999926ItX./( dJf(If666NIs482FItX./ y!!!  r8r>r}spacesrc fd|DS)Ncg|]}|zSrPrP)r@rBrs r6rEz_indent..s , , ,dFTM , , ,r8rP)r}rs `r6_indentrs , , , ,e , , ,,r8valuescjtd|DsJtd|DS)Nc3JK|]}t|tjVdSr.)rbrExprr@xs r6 zis_constants..s/CCqz!_122CCCCCCr8c3@K|]}t|tVdSr.rbrrs r6rzis_constants..s,771z!X&&777777r8)all)rs r6 is_constantsrsB CCFCCC C CCC C 77777 7 77r8c eZdZdZdedeeefdeeefdej de f dZ ded e e eeffd Z dd ed ed efdZdS) Environmenta:Environment of the scope Attributes: mode ('numpy' or 'cuda'): The rule for typecast. consts (dict): The dictionary with keys as the variable names and the values as the data that is determined at compile-time. params (dict): The dictionary of function arguments with keys as the variable names and the values as the Data. locals (dict): The dictionary with keys as the variable names and the values as the Data stored at the local scope of the function. ret_type (_cuda_types.TypeBase): The type of return value of the function. If it is initialized to be ``None``, the return type must be inferred until the end of transpilation of the function. generated (Generated): Generated CUDA functions. rrparamsrrcv||_||_||_i|_i|_||_||_d|_dS)Nr)rrrlocalsdeclsrrcount)r4rrrrrs r6r7zEnvironment.__init__s@   ') &(   " r8keyrc||jvr |j|S||jvr |j|S||jvr |j|SdSr.)rrr)r4rs r6 __getitem__zEnvironment.__getitem__$sU $+  ;s# # $+  ;s# # $+  ;s# #tr8r\prefixsuffixc||xjdz c_||j|}|||S|||S)Nr=)rget_fresh_variable_name)r4rrrgs r6rz#Environment.get_fresh_variable_name-sM a .$*.f.. : K//?? ?r8N)r\r\)rMrNrO__doc__r1rrrrTypeBaserr7rr rrrPr8r6rrs"h'dO  "*  ! "sxhn0E'F35@@@,/@9<@@@@@@r8rc  t||||||||S#t$r(} | } tr| |Yd} ~ nd} ~ wwxYw| |J)aTranspile the function Args: func (ast.FunctionDef): Target function. name (str): Function name. attributes (str): The attributes of target function. mode ('numpy' or 'cuda'): The rule for typecast. consts (dict): The dictionary with keys as variable names and values as concrete data object. in_types (list of _cuda_types.TypeBase): The types of arguments. ret_type (_cuda_types.TypeBase): The type of return value. Returns: code (str): The generated CUDA code. env (Environment): More details of analysis result of the function, which includes preambles, estimated return type and more. N)_transpile_function_internalr,rrL) rVrgrrrrrrrr5excs r6rr7s& + $ D& h ++ +      KK    KK5s A AA c td|D}t|tjs/t dt|t|j dkrgtj dkrW|j D]O}tj |tfddDstjddt P|j} | jt d t| jdkrt d | jt d t| jdkrt d d | jD} t| t|kr2t-|dt| dt|dtdt/| |D} t1||| ||} t3|jd| } ddt/| |D}d| jD}| jt<j| _|d| jd|d|d}tA||| z}tC|| fS)Nc6g|]\}}|t|fSrPrr@kvs r6rEz0_transpile_function_internal..Zs'@@@1Ax{{#@@@r8zNot supported: {}r)r c3 K|]}|vV dSr.rP)r@word deco_codes r6rz/_transpile_function_internal..dsDCC# 9,CCCCCCr8) rawkernel vectorizez Decorator z may not supported in JIT.z#`*args` is not supported currently.z4keyword only arguments are not supported currently .z&`**kwargs` is not supported currently.z+Default values are not supported currently.cg|] }|j SrP)arg)r@rs r6rEz0_transpile_function_internal..us . . .CG . . .r8z () takes z positional arguments but z were given.c8g|]\}}|t||fSrPrr@rts r6rEz0_transpile_function_internal..zs(CCCtq!AtAqzz?CCCr8T, c@g|]\}}||dSr.)declvarrs r6rEz0_transpile_function_internal..}s*MMMA!))At,,MMMr8cPg|]#\}}|j|ddz$S)N;)ctyper)r@nrs r6rEz0_transpile_function_internal..~s1OOOTQ!'//!T**S0OOOr8r^())"ritemsrbrXrzNotImplementedErrorformatr/rsrisys version_infounparseanywarningswarnRuntimeWarningrTvararg kwonlyargskwargdefaultsrpzipr_transpile_stmtsrhrHrrrvoidr r1)rVrgrrrrrrdeco argumentsrTrrrhparams_s local_varsheadr%rs @r6rrXs  @@@@@ A AF dCO , ,J!"5"<"r8oprTrct|r=tjt|}t |d|DSt |t jr|\}}tj ||}tj ||}t|j dr'|j |||}|tur|St|j dr'|j |||}|tur|St |t jr|\}}tj ||}tj ||}t|j dr'|j |||}|tur|St|j dr'|j |||}|tur|Stj|jt|}t)||d|S)Ncg|] }|j SrPobjrs r6rEz!_eval_operand..s 5 5 51 5 5 5r8_add_radd_sub_rsub)rr get_pyfuncr/rrbrXAddrinithasattrrrNotImplementedrSubrr get_ufuncr _call_ufunc)rrTrpyfuncryoutufuncs r6 _eval_operandrs T8 +DHH55 5 5 5 5 56777"cg 1 Ia   Ia   17F # # ',,sAq))C.(( 17G $ $ '--Q**C.(( "cg 1 Ia   Ia   17F # # ',,sAq))C.(( 17G $ $ '--Q**C.((  %chR 9 9E udD# . ..r8rdtypec t||jkrtdg}|D]}t|tr%t j|j|jj }nDt|j tj r |j j }ntd|j d||t j|t!|||j}|td|jd||j||jdkr|jdrtj |jd}|jdd } g} t1||jD]S\}}t5|tj |d |}t7j||}| |Td } t;|jD]^} tt=t?j d !| |jdkrd } d| d|jvrd } _|j"#|j$| r]tK| D]#\} }| d| |j&} $d| dtO|zdz} nd(dt;|jD} |jdtOtSj |jd}d(dt;|jD}d| d|d|d|d|d| d }|j"#|d(d| D}|d|d} t7| |StUd|jd) Nzinvalid number of argumentsz cupy.ufunc: z is unsupported"z(" does not support for the input types: r=zout0 = rr\r!Tzin{}Fin_typer out0_typerrcg|]}d|d S)z typename inrrPr@rAs r6rEz_call_ufunc..s7,B,B,B+,&a&&&,B,B,Br8r~cg|] }d|d| S)rz_type inrPrs r6rEz_call_ufunc..s)NNN 3Q 3 3 3 3NNNr8z template __device__ r^z) { typedef z out0_type; return z; } cg|] }|j SrPr%)r@as r6rEz_call_ufunc..s'B'B'B1'B'B'Br8zufunc `z` is not supported.)+rsninrmrbrrget_ctype_from_scalarrrrrrScalarrpr guess_routinerrg error_funcnoutroutine startswith out_typesr_rr_astype_scalarrrrangelistrefinditerrrr _preamblerIr%r1rHnumpyr)rrTrrrrrrout_typeexpr in_paramscan_use_inline_expansionrAtemplate_typenames ufunc_namer ufunc_codein_params_codes r6r r s7  4yyEI6777H  a " " E5chFFLAA !3 4 4 E AAC17CCCDD D  &ueHoouch O OB z N N NH N NPP P }   zQ2:00;;%bl1o66z!!)R00 bk**  DAqq+"4Q"7"7cJJA !S!!A   Q    #' uy!! 1 1A4 GNN1$5$5rzBBCCDDqHH+0(A}}} **+0( u/// # 5!),, 6 61||HHHaf55k3x==AAACGDD!%,B,B05ei0@0@,B,B,B"C"C !JLLU[a-I-I)J)JLLJYYNNU59=M=MNNNOOF(: !$*  J M " ": . . .!YY'B'B 'B'B'BCCN 44>444DD(### G GGG H HHr8stmts is_toplevelc\g}|D]&}|t|||'|Sr.)extend_transpile_stmt)r7r8r codeblocksstmts r6rrsC JCC/$ SAABBBB r8r=ct|tjrtdt|tjtjfrtdt|tjrt|j|}t|tr |j dgStj ||}|j }|j||_n%|j|krtd|jd|d|jdgSt|tjrtd t|tjrt'|jd krtd t|j|}|jd }t+|rt|tjrl|j}t|j t0sK|r:t||trt3d |d||j|<gSt3dtj ||}t7||||St|tjrt|j|}t|j|}t|tst3d|jtj ||}t|d|j }t?|j ||f|} t|tsJt| tsJt|j tBj"sJt| j tBj"sJtG| j j$|j j$dd|j %d|jz|zdz|j &|| zdzgSt|tj'rt'|j(d krtdt|jtjsJ|jj}t|j)|} ||} | 4t|jj| j }||j*|<||j+|<nVt| trt3d| j | j kr"t3d|d| j d| j t| tXj-stdt]|j/d|} | j d| j0jd| j1jd | j2j} d!}| j3d"urd#}n | j3durd$}d%| d|d&}ti||d'g| zg}| j5}|d"urd(g|z}n | d)|d*g|z}|St|tj6rtd+t|tj7rt'|j(d krtdt|j8|}ts|tBj:d,|}tj ||}t]|j/d|} d-|jd*}ti|| gSt|tj;rt|j8|}t+|r&|j r|j/n|j(}t]|||Sd.|jd*}t]|j/d|}t]|j(d|}ti||tid/|gSt|tj<tj=frtd0t|tj>tj?frtd1t|tj@r9t|j8|}t+|r |j sJdgSd2|zd3zgSt|tjAtjBfrtd4t|tjCtjDfrtd5t|tjEr2t|j|}t+|rdgn |jdzgSt|tjFrdgSt|tjGrtd t|tjHrtd J)6zaTranspile the statement. Returns (list of [CodeBlock or str]): The generated CUDA code. z!class is not supported currently.z-Nested functions are not supported currently.Nzreturn;z!Failed to infer the return type: z or zreturn rz!`del` is not supported currently.r=Not implemented.rzType mismatch of variable: ``z.Cannot assign constant value not at top-level.zCannot augassign to _tmp_r!z{ &z; z; }zwhile-else is not supported.z'loop counter must not be constant value!Data type mismatch of variable: ``:  != z.for-loop is supported only for range iterator.Fz __it = z , __stop = z , __step = z+__step >= 0 ? __it < __stop : __it > __stopTz __it < __stopz __it > __stopzfor (z; __it += __step)z = __it;z#pragma unrollz#pragma unroll(rz`async for` is not allowed.r"zwhile (zif (elsez#Switching contexts are not allowed.zthrow/catch are not allowed.zassert(z);z0Cannot import modules from the target functions.z3Cannot use global/nonlocal in the target functions.)IrbrXClassDefrrzAsyncFunctionDefr{_transpile_exprvaluerrrrrrrmr%DeleteAssignrstargetsrNameid _typeclassesrpr_transpile_assign_stmt AugAssigntargetrrrrr!rrrassignFororelseiterrrrRangerrhrDstopstepstep_is_positiver unrollAsyncForWhiletestr(bool_IfWith AsyncWithRaiseTryAssertImport ImportFromGlobalNonlocalrPassBreakContinue)r=r8rrJrvarrgrStmpritersloop_varrh init_codecondrr%r\ conditionr7 then_body else_bodys r6r;r;se $ %%G!"EFFF$#*>?@@=! ;== =$ ##) C00 eX & & 59+<;  %%% K < CLL \Q  ICLIIaIIKK K'%*'''(($ ##G!"EFFF$ ##D t|   ! !%&899 9 C00l1o    J:c38#<#< J6Dei66 JJ!#d)T22P'(Nt(N(N(NOOO',CJt$I#HJJJ %%%%c3{CCC$ &&: C00 c22&$'' B@6;@@AA A %%%3..w77FFtwe c::&$'''''&$'''''&, (:;;;;;&, (:;;;;; L  2K A A Av|++C#(NFCCCdJ ##C001389: :$  + t{  a  %&DEE E$+sx00000{~ 3//t9  $+.%+66F%CJt $CIdOO ( + + 6EFF F ^u{ * *5D55>55',{5566 6%!677 B%@BB B  5#66 33${/33!&33"'33 =  !T ) )"DD  #u , ,"D;y;;D;;;$Tt,=,=,=+>+EFFG T>>$%,DD  /f///047D $ %%86777$ ""' t{  a  %&DEE E#DIs33 "9k.?3OO Ii--  5#66****$%%&&$J#DIs33  " " =!*?DIIDKE#E;<< <'in'''$TYs;; $T[%== $ **Ifi,H,HII$3=122@>???$CG,--97888$ ##. 3//    .9  95L%,- -$S^455MKLLL$S\233PNOOO$!!D C00$U++Cuu%*s2B1CC$!!u $ ""6!"4555$ %%6!"45555r8r0ct||}t|tr&t|jtjr|jS|S)z`Transpile the statement. Returns (Data): The CUDA code and its type of the expression. )_transpile_expr_internalrbrrrr)r0rress r6rIrIsG #4 - -C#x  Z9M%N%Nw r8c t|tjrBfd|jD}|d}|ddD]}t |j||f}|St|tjrBt|j}t|j }t |j||fSt|tj r,t|j }t |j|fSt|tj rtdt|tjr\|jg|jz}t!|dkrtdfd|D}t |jd|St|tjr2t|j}t|j}t|j} t|t,r |jr|n| S|jjjdkrt7d t9|| t9| |} }|jj| jjkr)t7d |jjd | jjt;|t<jd }tAd |j!d|j!d| j!d|jSt|tj"rrt|j#} fd|j$D} i} |j%D](} | j&Jt| j'| | j&<)tPj)}tU| r| j|vr || j} t| tVj,r| j-g| Ri| St| t,st7d| d| j} t| t\jj/r| j0g| Ri| St| t\j1r| j2st7d| j3j4dfd| D} tkd| D}tm| j3dgj7|dj8\}}d9d| D}tA|d |d|St| ttj;r| <dt-dj}t!| dkr6t{t}| }t7d|d| j?t| | |StUg| | RrLd| D} td| BD} t-| | i| StjD| rgt| trRt!| dkrt7d| d t=jG| }t;| d|d St7d!| j4d"|t-dSt|tjrt-|j'St|tjHr;t|j'}t|jI}t||St|tjKr(|jL}|td#|jL|St|tjNrt|j'}t|t,r't-t|j|jPSt|t@rqt|j|jPrWt|j|jPd}t|tjSr ||St|t@r|Std$|jPt|tjUrfd%|jVD}td&|Dr&t-tkd'|DSfd(|D}d9d)|D}t!|dkr0tAd*|dt=jUd+|DStAd,|dt=jUd-|DSt|tjXrt|j'Std.Zt|)/Nc0g|]}t|SrPrIr@r5rs r6rEz,_transpile_expr_internal..s#???a/!S))???r8rr=r?z2Comparison of 3 or more values is not implemented.c0g|]}t|SrPr|r}s r6rEz,_transpile_expr_internal..s#:::a/!S)):::r8cz/Complex type value cannot be boolean condition.z*Type mismatch in conditional expression.: rEr"rz ? z : rc0g|]}t|SrPr|r@rrs r6rEz,_transpile_expr_internal..#;;;A3'';;;r8'z' is not callable.zCalling __global__ function z) from __global__ function is not allowed.c:g|]}tj|SrPrrrs r6rEz,_transpile_expr_internal..s%444!DIa%%444r8cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..s444!ag444r8 __device__rcg|] }|j SrPrrs r6rEz,_transpile_expr_internal..s"8"8"8a16"8"8"8r8rz!' is an invalid keyword to ufunc cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..s(((aAE(((r8c&g|]\}}||jfSrPrrs r6rEz,_transpile_expr_internal..s"AAA$!QAqu:AAAr8zfunction takes z invalid number of argumentzInvalid function call 'z'.zUnbound name: zUnknown attribute: c0g|]}t|SrPr|rs r6rEz,_transpile_expr_internal..8rr8c8g|]}t|tSrPrrs r6rEz,_transpile_expr_internal..:s"666A 1h''666r8cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..;s"7"7"7Q15"7"7"7r8c:g|]}tj|SrPrrs r6rEz,_transpile_expr_internal..=s%000a !S!!000r8cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..>s444!qv444r8zSTD::make_pair(cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..C"9"9"9q17"9"9"9r8zSTD::make_tuple(cg|] }|j SrPrrs r6rEz,_transpile_expr_internal..Grr8zNot supported: type {})\rbrXBoolOprrrBinOprIleftrightUnaryOpoperandrcrCompare comparatorsrsopsIfExpr_rhrVrrrrkindrp _infer_typer(rr`rr%CallrVrTkeywordsrrJrbuiltin_functions_dictrr BuiltinFunccallrr _instantiate _JitRawKernel_device_funcrMrrrrrHrrpopnextrWrgr rrrnisclass issubclassrPr! Subscriptslice _indexingrNrO NameError AttributerGattrr types MethodTypeAttributeErrorreltsrIndexrmrr/)r0rrrJrhsrrrsrrrVrTrUkw builtin_funcsrfnamer&r1rrgrarrayindexrr elts_codes ` r6rxrxs $ ##????4;???q !"": > >C!$'E3<==EE $ "":ty#.. C00TWtUmS999$ $$5 c22TWuh444$ ##6!"4555$ $$7)t// v;;!  %DFF F::::6:::TXa[&#666$ ""Ety#.. DIs + + DK - - dH % % ('11a ' :  C ' 'MNN N1a%%{1a'='=1 7=AGM ) )67=66&'gm6677 7dK$5xEE: ::af:::::AGDDD$!!;Ety#..;;;;;;;35- < < @?4:3F???@@@5444t444D44t44455H!4 \NCH$ "/"/ E; "8"84"8"8"899I5//9///== = dGM * * 7JJw77;E6{{QDLL))JJJtyJJLLLtT5#66 6  0 0  0 0 0 3((4(((DAA&,,..AAABBFDD$1&1122 2 ?4  AZl%C%C A4yyA~~GdGGGIII&t,,E!$q'5(C@@ @C$-CCCDDD |~~$ %%$ ###$ &&, C00 C00s+++$!!DG  =6TW6677 7 $ && @ C00 eX & & ;GEIty99:: : eT " " wu{DI'F'F 5; 488D$ 011 #tE{{"$%%  >49>>???$ ""<;;;;;;; 66666 7 7 :E"7"7$"7"7"78899 900004000II44t44455 t99>>.)...!"9"9D"9"9"9::<< </9///!"9"9D"9"9"9::<< <$ ""0tz3/// -44T$ZZ@@ A AAr8lvaluervaluect|trtdt|jtjr7t|jtjrt ||jd|}n7|j|jkr'td|jd|jd|j|j||dzgS)Nz/lvalue of assignment must not be constant valuer!rCrDrEr) rbrrprrr!r(r%rT)rrrs r6_emit_assign_stmtrPs &(##KIJJJ6<!34416<);<<1 k3GG  % % 0  0 0| 0 0!' 0 011 1 L   / /# 5 66r8rSrJdepthc t|tjrr|j}||}|Pt ||j}||j|<|r%|dkr|j||dzgS||j|<t|||St|tj r4t j t|||}t|||St|tj rNt|jtj st|jdt!|j}t!|jj|krtd|dt!|jj|krtd|d|jd||dzg}t'|D]a} d| d |d} |jj| } t)|j| |t | | ||d z} || bt-d |gSJ) Nrrz cannot be unpackz$too many values to unpack (expected rz¬ enough values to unpack (expected _temp STD::get(_tempr=r\)rbrXrNrOrrrrrrrrrIrrrmrsrrr)rQr:r ) rSrrJr8rrgrsizerrAr%rr=s r6rQrQcsO&#(## 5yT >$ ,,F%CJt  @uzz ++D%883>??$CIdO 444&#-((5?6377== 444&#)$$&%+{'899 @ >>>?? ?6; u{ ! !D ( (KDKKKLL L u{ ! !D ( (MdMMMNN N$$_U__e<(rz.Tuple is not subscriptable with non-constants.rz1Scalar indexing is not supported for 0-dim array.r=z ._slicing(uizArray indices must be integers.[]z+The number of indices is beyond array dim: zArray indices must be scalar.iuzArray indices must be integer.z, Dim()z[0]z [STD::get<0>(z)]z ._indexing(z#Advanced indexing is not supported.Fz is not subscriptable.)rbrrrpr/rrrrrrr% ArrayBase_ndimr!rCArray _c_contiguous_index_32_bitsr child_typers IndexError) rrrrArndim index_dtype new_carrayrs r6rrsd %""K eX & & 2EIei011 1EI I I IKK K IeS ! !E%+{011J eX & & ; A !!$A6A66666:: :HIII%+{4552 %%%{  ek;#5 6 6 G++KqyyGIIIaxx(/K%K%)K-K. 00 z::UZ:::JHHHt++ ABBB:-- ---u{/EGG G ek;#4 5 5 (c%++,,,, H$HHJJJ[& F F!![%788E#$CDDD7(>>K-K. 00 "JIIc%+2C.D.DIIIuz>>V>>> KKKqyyz&&& (>@@@qyyz>> >>>K*,,,:77%*777 &(( ( ek;#5 6 6 CABB Bu uz999 : ::r8_Trrcastingc t|trCt|trJt|j|jSt|jtjst|j d|jj}|j}||kr|Stj |d||std|d|d|d|j dkrL|j dkrA|j dkrtjd t td |d |j d |Std |d |j d |S)Nz is not scalar type.rzCannot cast from 'z' to z with casting rule .rbz:Casting complex values to real discards the imaginary partrz)(z.real())r)rbrrrr/rrrr!rpr%r.can_castrrrr )rrrrfrom_tto_ts r6r(r(s !X1a&&&&& ((//000 ag{1 2 2916777888 W]F ;D ~~ >&++a..$ 8 8- , , ,d , ,!( , , ,-- -{cdi3.. 9   ML   1111115999 &E&&QV&&& . ..r8hintcXt|trt|jtjrt j||St j||}t|jtj sJt||jd|}t j||S)Nr!) rbrrr.genericrrrrr!r()rrrcast_xs r6rrs a " "!j &F&F!yC   9T3  D dj+"4 5 555 5 Atz; < )r)TrX collectionsrnrwnumbersr+rtypingrrrrrrr r rrr.cupy.exceptionsr cupy_backends.cuda.apir cupy._core._codeblockr r cupy._corercupy._core._dtypercupyxr cupyx.jitrrrcupyx.jit._internal_typesrrrrrboolr`NumberrPrr _CastingTyper1 namedtupler#rSr,rZrrrrrrrrrrrYrrrr r=rr;r0rIrxrintrQrrr!r(rrPr8r6rs  MMMMMMMMMMMMMMMMMMMM ************66666666444444!!!!!!%%%%%%%%%%%%******......$$$$$$ ek7>2  S <=?LLL      : : : : :y : : :   HHHV!M!M!M!M!M!M!M!MHOOO:6--49-c-T#Y----8/.848888 3@3@3@3@3@3@3@3@lB---`$/ G$/uXt^,-$/$/ 8T> $/$/$/$/NDI}DIuXt^,-DI $DI DI  DIDIDIDINCH~ bhbbb bbbbJ #(  9M    XBsx XB XBXBXBXBXBv7hn%777 77770 ((( (( (  (  ((((VI;#I;#I;I; 4> I;I;I;I;X WT8T54#899/ /!// /  ////D " 4  "HdN# " " " " " " " "r8