o i @sddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZe ZejZeedZed d Zeejeejd d Zeejd dZeejddZeeeejddZ dS))singledispatch)ir)typescgutils) NumbaWarning)Registry) nvvmutils)warncCstd|f)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 %s)NotImplementedError)tycontextbuildervalrQ/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/cuda/printimpl.py print_itemsrcCs:|tjvr d}tj}nd}tj}|||||}||gfS)Nz%lluz%lld)runsigned_domainuint64int64cast)r r rrrawfmtdsttypelldrrrint_print_impls  rcCs||||tj}d|gfS)Nz%f)rrfloat64)r r rrrrrrreal_print_impl)s rcCs.|j}t|ts Jd}|||}||gfS)Nz%s) literal_value isinstancestrinsert_string_const_addrspace)r r rsigvalpyvalrrrrrconst_print_impl/s   r#cCst|j}g}g}tt|j|D]\}\}} t|||| \} } || || qd |d} t |dkrFd} t | t | dd} ||| }t||}t||}t|j}|||||tf|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%%)rdeclare_vprintmodule enumeratezipargsrappendextendjoinlenr rreplacer rmake_anonymous_structalloca_once_valuecallbitcastvoidptrget_dummy_value)r rsigr,vprintformatsvaluesiargtypeargvalargfmtargvalsrmsgfmtarrayarrayptrrrr print_varargs8s$          rEN)! functoolsrllvmliter numba.corerrnumba.core.errorsrnumba.core.imputilsr numba.cudarwarningsr registrylower PointerTypeIntTyper6rregisterIntegerIntegerLiteralrFloatr StringLiteralr#printVarArgAnyrErrrrs*