o ij @svdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddlmZddlmZmZmZmZddlmZddlmZdd lmZmZdd lmZmZmZmZej j!Z"e#ed Z$e#ed Z%ej j&Z'ej j(Z)e%ej j*Z+ej,j-d kr~d ndZ.dZ/e/Z0ej1j2Z3ej j4Z5ej j6Z7ddZ8ee8ddZ9ee:ddZ;eddZej>e%ej>ddddZ?eddZ@eddZAee%e%e%e%e%e%ejBejBd dd!d"ZCeejDd#eejEd#d$d%ZFeejGd#d&d'ZHeejId#d(d)ZJe%j-d*d+kr$e%d,ZKe%d-ZLe%d.ZMed/ejNidd0d1ZOne%d2ZKe%d3ZLe%d4ZMed/ejNidd5d1ZOee%e%e%e%e%d6dd7d8ZPeejQd#d9d:ZRdd;lmSZSmTZTmUZUmVZVmWZWmXZXGdd?d?eWZZGd@dAdAeWZ[GdBdCdCeWZ\GdDdEdEeXZ]edFdGdHgZ^dIdJZ_e_Z`e7dKvr e7dLkrdMZaebeaeejNejNdNddOdPZceejNejNejNejNejNejNdQddRdSZdeejNejNejNejNdTddUdVZeeejNejNejNejNdTddWdXZfdYdZZgegd[Zhegd\Zie7d[krehneiZjnd]e7Zaekeaed^d_Zld`daZmeemdbdcZnedde%iddedfZoeejpd#dgdhZqdS)iz& Hash implementations for Numba types N) namedtuple)ir)literal_unroll)overloadoverload_method intrinsicregister_jitable)errors)types) grab_byte grab_uint64_t) const_int get_next_intget_next_int32 get_state_ptrpy_int =iCBcCsdSN) hash_funcrrV/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/cpython/new_hashing.py _defer_hash/srcsd|dfdd}|S)Nzunhashable type: ''cs|durt|Sr) TypeError)objrerr_msgrrimpl7szol_defer_hash..implr)rrrrrr ol_defer_hash3s  r csd|dfdd}|S)Nz+No __hash__ is defined for object of type 'zn' and a generic hash() cannot be performed as there is no suitable object represention in Numba compiled code!cs"t|dr t|t|dSt)N__hash__)hasattrrgetattrr)rattempt_generic_msgrrrGs zhash_overload..implr)rrrr$r hash_overload@s  r&cCs t|}|tdkrtd}|S)N) _Py_hash_tint)valasintrrrprocess_returnOs r-)xymesign_PyHASH_MODULUS _PyHASH_BITSlocalscCst|st|r|dkrtSt St}t|St|\}}d}|dkr,d}| }d}|rZ|d>t@|t d?B}|d9}|d8}t |}||8}||7}|tkrX|t8}|s0|dkrc|t }n t dd|t }||>t@|t |?B}||}t|S)Nrr'gA) npisfiniteisinf _PyHASH_INF_prng_random_hashr-mathfrexpr3r4r*)vr.r0r1r2r/rrr_Py_HashDouble_s:  rAcCsdd}ttj}||fS)NcSs|d}||tS)Nr)fpextr DoubleType)cgctxbuilder signatureargsr+rrrrsz_fpext..impl)r c_float64 c_float32)tyctxr+rsigrrr_fpexts rLcCsdd}t}||fS)NcSs@t||d}tt}tdkrt|||}|St||||d}|S)NinternalrF)rr _hash_widthrr)rDrErFrG state_ptrbitsvaluerrrrs  z_prng_random_hash..impl)r))rJrrKrrrr=sr=)r.p1p2p3p4r3r4 _PyLong_SHIFTc Csdt}td|?}dtd}d}tt}t|dddD]'}|t>}|t@}||?} || B}|t||t?|@7}|tkrG|t8}q t|S)Nrr@r7r')rVr c_uint32r4ranger3r)) r+ _tmp_shift mask_shiftir.rTidxrRrSrUrrr _long_impls r^r!csHtjdkrdndtdt|ddrtjntjfdd}|S) Nlr(lsignedFcs|}t|}|tkr*|dkrd}t|S|kr"t}t|St|}t|Sd}|dkr5| }d}t|}|r>| }t|S)NrFT)absr3r)r^r-)r+magret needs_negate_BIG _HASH_I64_MIN _SIGNED_MINrrrs&    zint_hash..impl)sysmaxsizer c_int64r#c_uint64r+rrrerint_hashs  rncCs"|jdkr dd}|Sdd}|S)NrWcSs t|}|Sr)rA)r+hashedrrrrszfloat_hash..implcSstt|}t|}|Sr)r9float64rLrA)r+ fpextendedrorrrrs)bitwidthrmrrr float_hashs  rscC dd}|S)NcSs(t|j}t|j}|t|}t|Sr)hashrealimag _PyHASH_IMAGr-)r+hashrealhashimagcombinedrrrr%s   zcomplex_hash..implrrmrrr complex_hash#s r|lJ fq lOkO8 lg,,>r.cC|td>|td?BS)Nr!r np_uint64r.rrr_PyHASH_XXROTATE>rlyn<lwJ igVcCr)N rrrrrrGr)acclane_PyHASH_XXPRIME_5_PyHASH_XXPRIME_1tlcCst|}t}t|D]}t|}|tdkrdS||t7}t|}|t9}q ||ttdAA7}|tdkr.implrrmrrr tuple_hashdr)c_size_tc_ubyterl pythonapi StructureUnionc@eZdZdefdefgZdS)FNVprefixsuffixN)__name__ __module__ __qualname__r_fields_rrrrrrc@r)SIPHASHk0k1N)rrrrlrrrrrrrrc@ eZdZdedfdefgZdS)DJBX33ApaddingrNrrrrrrrrrrr rc@r)EXPATrrhashsaltNrrrrrrrrc@s2eZdZdedfdefdefdefdefgZdS)_Py_HashSecret_tucfnvsiphashdjbx33aexpatN) rrrrrrrrrrrrrrs r_hashsecret_entrysymbolrQcsJttd}ifdd}|d|jj|d|jj|d|jjS)zRead hash secret from the Python process Returns ------- info : dict - keys are "djbx33a_suffix", "siphash_k0", siphash_k1". - values are the namedtuple[symbol:str, value:int] _Py_HashSecretcs>d|}t|}t|}t||t||d|<dS)Nz_numba_hashsecret_{})rrQ)formatctypesrl addressofll add_symbolr)namer+ symbol_nameaddrinforrinjects    z!_build_hashsecret..injectdjbx33a_suffix siphash_k0 siphash_k1)rin_dllrrrrrr) pyhashsecretrrrr_build_hashsecrets  r) siphash13 siphash24rraFNV hashing is not implemented in Numba. See PEP 456 https://www.python.org/dev/peps/pep-0456/ for rationale over not using FNV. Numba will continue to work, but hashes for built in types will be computed using siphash24. This will permit e.g. dictionaries to continue to behave as expected, however anything relying on the value of the hash opposed to hash as a derived property is likely to not work as expected.r.bcCs t||>|td|?BS)NrWrrrrr_ROTATEs rarcdstcCsB||7}||7}t|||A}t|||A}t|d}||||fS)Nr)rrrrr _HALF_ROUNDs   rv0v1v2v3cCs@t||||dd\}}}}t||||dd\}}}}||||fS)Nrr)rrrrr _SINGLE_ROUND#s rcCs8t||||\}}}}t||||\}}}}||||fSr)rrrrr _DOUBLE_ROUND,s rc sr|dkr tdn |dkrtdnJdttjtjtjtjtjtjtjtjtjtjd dfd d }|S) NrTrFr unreachable) rrrrrmirmaskjmpohexefefr5cst|d>}|tdA}|tdA}|tdA}|tdA}d} |dkrMt|| } | d7} |d8}|| N}||||\}}}}|| N}|dks)td} | d} td } |d kryd }t| |>}| |@tt|| d |>B} |d krd }t| |>}| |@tt|| d|>B} |dkrd}t| |>}| |@tt|| d|>B} |dkr| tdM} tdD]}|d}t| |>}| |@tt|| ||>B} q|dkrd}t| |>}| |@tt|| d|>B} |dkr!d}t| |>}| |@tt|| d|>B} |dkr7| }| |@tt|| dB} || O}||N}||||\}}}}||N}|| N}||||\}}}}||||\}}}}rv||||\}}}}||A||AA} | S)N8lue`5{lmo\I{#lar\clser+#rr}r70(rr~lr)r rr r rY)rrsrcsrc_szrrrrrr]rrboffsetrrrr\ _EXTRA_ROUND_ROUNDERrr_siphash?s        z_gen_siphash.._siphash)rrrr r)algrrrr _gen_siphash5s( Errrz'Unsupported hashing algorithm in use %scsDt|tjs tdt|jjtj}||}fdd}||fS)z&Emit code to load the hashsecret. zrequires literal stringcsL|j}z|}Wntytj|tdd}Ynw||}|S)NrWr)module get_globalKeyErrorrGlobalVariableIntTypeload)rDrErKrGmodgvr@symrrrs  z%_inject_hashsecret_read..impl) isinstancer StringLiteralr TypingError _hashsecret literal_valuerr)rJrrestyrKrrrr_inject_hashsecret_reads     rcCs t|jSr)rrQrrrr_load_hashsecrets rcCrt)NcSrr)rrrrrimprz"_impl_load_hashsecret..impr)rr rrr_impl_load_hashsecretrr _hashcCs|dkrtdS|tkr3td}t|D]}|d>|tt||}q||N}|tdN}t|Stt tdt td||}t|}t|S)Nrirrrr) r-_Py_HASH_CUTOFFrrYr9uint8r r _siphasherr r)r+_lenr r]tmprrr _Py_HashBytess   rcsddlmfdd}|S)Nr_kind_to_byte_widthcs4|j}t|}|j}|dkr|St|j||S)Nr')_kindrr r_data)r+ kindwidthr current_hashrrrrs zunicode_hash..impl)numba.cpython.unicoderrmrrr unicode_hashs   r)r__doc__r>numpyr9rirwarnings collectionsrllvmlite.bindingbindingrllvmliternumbarnumba.core.extendingrrrr numba.corer r numba.core.unsafe.bytesr r numba.cpython.randomimplr rrr hash_infowidthrNr#r)rinfr<nan _PyHASH_NANmodulusr3rrrr4_PyHASH_MULTIPLIERrxint_infobits_per_digitrVcutoffr  algorithm_Py_hashfunc_namerr rur&r- np_doublenp_intcrArLr=c_int32r^IntegerBooleanrnFloatrsComplexr|rrrrrr BaseTuplerrrrlrrrrrrrrrrrmsgwarnrrrrr _siphash13 _siphash24r ValueErrorrrr r UnicodeTyperrrrrs             .       ,             /    [