i *ddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZe Zej Zej"ej$dZed Zej+ej,ej+ej.d Zej+ej2d Zej+ej6d Zeeej<ej>d Z y))singledispatch)ir)typescgutils) NumbaWarning)Registry) nvvmutils)warnctd|)z Handle printing of a single value of the given Numba type. A (format string, [list of arguments]) is returned that will allow forming the final printf()-like call. z*printing unimplemented for values of type )NotImplementedError)tycontextbuildervals i/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/cuda/printimpl.py print_itemrs !#& ''c|tjvrd}tj}nd}tj}|j ||||}||gfS)Nz%lluz%lld)runsigned_domainuint64int64cast)rrrrrawfmtdsttypellds rint_print_implrsN U " "",,++ ,,wR 1C C5=rcP|j|||tj}d|gfS)Nz%f)rrfloat64)rrrrrs rreal_print_implr )s& ,,wR 7C #;rcp|j}t|tsJd}|j||}||gfS)Nz%s) literal_value isinstancestrinsert_string_const_addrspace)rrrsigvalpyvalrrs rconst_print_implr(/s@   E eS !! ! F  / / ?C C5=rctj|j}g}g}tt |j |D];\}\}} t |||| \} } |j| |j| =dj|dz} t|dkDr$d} t| t| jdd} |j|| }tj ||}tj"||}tj|j}|j%|||j'|t(f|j+S)zThis function is a generic 'print' wrapper for arbitrary types. It dispatches to the appropriate 'print' implementations above depending on the detected real types in the signature.  zjCUDA print() cannot print more than 32 items. The raw format string will be emitted by the kernel instead.%z%%)r declare_vprintmodule enumeratezipargsrappendextendjoinlenr rreplacer%rmake_anonymous_structalloca_once_valuecallbitcastvoidptrget_dummy_value)rrsigr2vprintformatsvaluesiargtypeargvalargfmtargvalsrmsgfmtarrayarrayptrs r print_varargsrK8s+  % %gnn 5FG F )#chh*= > GV$WgwGv g!? XXg  %F 4y2~N S,T*  / / @C  ) )'6 :E((%8H  % %gnn 5F LL#wxABC  " " $$rN)! functoolsrllvmliter numba.corerrnumba.core.errorsrnumba.core.imputilsr numba.cudar warningsr registrylower PointerTypeIntTyper<rregisterIntegerIntegerLiteralrFloatr StringLiteralr(printVarArgAnyrKrrr`s$%*(  : "..A ' '' U]]# U))*+$ U[[!"  U(()*ulell599%&%'%r