j'rddlmZddlZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z  ejdkZejej dkZejdvZdd Zdd Zgd ZgdZgdZdZdZdZdZGddejZGddZ ddZ!d dZ"dS)!) annotationsN) CFUNCTYPEPOINTERc_longc_size_tc_uint32c_ulongc_void_p)ClassVarnt>i686-64x64x86i368i686amd64x86_64returnstrcddlm}|jSNr) polars._plr_plr RUNTIME_REPR)plrs V/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_cpu_check.pyget_runtime_reprr5s   ctypes.CDLLc tjjdkrtjddStjddS#t $r2ddlm}tj|ddcYSwxYw) NDarwinz libc.dylibT) use_errnoz libc.so.6r)utilc)osunamesysnamectypesCDLL Exceptionr$ find_library)ctutils r_open_posix_libcr.;sE 8::  ) );|t<<< <;{d;;; ; EEE)))))){6..s33tDDDDDDEs1A A 9BB)Sr0r0r0_r0Or r0W [)r/r0Ir0Dr0r3r4Ar0rBr0Yr7rBr0r>r rBr0Qr:r;r<)r/r9|$r:rFr@rHrFLrHr3r4r0r5r0r6r7r0r8r r0r9r:r6r;r< i@c2eZdZUddDZded<dS) CPUID_structc g|] }|tf S)r).0rs r zCPUID_struct.s)222H 222r)eaxebxecxedxz ClassVar[list[tuple[str, type]]]_fields_N)__name__ __module__ __qualname__rZ__annotations__rRrrrPrPsH22;222HrrPc(eZdZd dZd dd Zd d Zd S)CPUIDrNonec:trFtr!tjd|_t }n3tjj|_t}ntrtnt}t|}tj |z|}trt|jj _tj tjtjtjg|jj _|j d|t$t&zt(|_|jsd}t-|tj|j||nddl}t3}|j}tj tjtjf|_tj|_|d||j|jz|j|jz|_tj tj !|j|_|j"|||j||j|j#zdkrd}tI|tKdtMtNtPtP}||j|_)dS)Nz kernel32.dllz)could not allocate memory for CPUID checkrz*could not execute mprotect for CPUID check)* _IS_WINDOWS _IS_64BITr)r*win_WINDOWS_64_OPCwindllkernel32 _CDECL_32_OPC _POSIX_64_OPClenc_ubyter VirtualAllocrestyperr argtypes _MEM_COMMIT _MEM_RESERVE_PAGE_EXECUTE_READWRITEaddr MemoryErrormemmovemmapr.mprotectc_int MAP_PRIVATE MAP_ANONYMOUS PROT_READ PROT_WRITE addressof from_bufferwrite PROT_EXEC RuntimeErrorrrrPrfunc_ptr) selfopcsizecodemsgrwlibcrx func_types r__init__zCPUID.__init__s  @ $";~66% "=1##,?---C3xx%, $ (,4DH ! ) .DH ! * --dK,68ODI9 'A!#&&& N49dD 1 1 1 1 KKK$%%D}H!'&/6< PH %|H   4#550 DI ()D)DTY)O)OPPDI IOOD ! ! !x 4$.)HIIQNNB"3'''dGL$9$98XNN ! $),, rrrVintrXrPcPt}|||||S)N)rPr)rrVrXstructs r__call__zCPUID.__call__s& fc3''' rctrat|jj_t t tg|jj_|j|j dtdSdSr) rdrrf VirtualFreeror rr rprt _MEM_RELEASE)rs r__del__z CPUID.__del__sW  =+1DH (-5x,IDH ) H A| < < < < < = =rN)rra)r)rVrrXrrrP)r[r\r]rrrrRrrr`r`sV;-;-;-;-z ======rr`dict[str, bool]ctsiSt}|dd}|dd}|dd}t|jdzt|jdzt|jdzt|jdzt|jdzt|jd zt|jd zt|jd zt|jd zt|jd zt|jdzt|jdzt|jdzt|jdzdS)NrCrr5lirLrMiii@iir  )sse3ssse3fma cmpxchg16bzsse4.1zsse4.2movbepopcnt pclmulqdqavxbmi1bmi2avx2lzcnt)_SUPPORTS_CPUIDr`boolrXrW)cpuidcpuid1cpuid7cpuid81hs r_read_cpu_flagsrsV   GGE U1a[[F U1a[[FuZ##HVZ6*++fjF+,,FJ'*++6:122vzW-..vzW-..fjG,--vzW-..&*/00FJ'*++VZ6*++VZ6*++VZ6*++hlf-..  r feature_flagsracd|dD}d|D}|rtjdrdSt }g}|D]7}||vrd|}t |||s||8|r:ddl}|dd |d td dSdS) Ncbg|],}|d|d-S)-+) startswithlstriprSfs rrUz#check_cpu_flags..sCq||C?P?P  r,c"g|] }|r|dk | S)z ctr-staticrRrs rrUz#check_cpu_flags..s7 l"" """rPOLARS_SKIP_CPU_CHECKzunknown feature flag: rz[Missing required CPU features. The following required CPU features were not detected: z, a  Continuing to use this version of Polars on this processor will likely result in a crash. Install `polars[rtcompat]` instead of `polars` to run Polars with better compatibility. Hint: If you are on an Apple ARM machine (e.g. M1) this is likely due to running Python under Rosetta. It is recommended to install a native version of Python that does not run under Rosetta x86-64 emulation. If you believe this warning to be a false positive, you can set the `POLARS_SKIP_CPU_CHECK` environment variable to bypass this check. rC) stacklevel) splitr&environgetrrappendwarningswarnjoinRuntimeWarning)rexpected_cpu_flagssupported_cpu_flagsmissing_featuresrrrs rcheck_cpu_flagsrsK,22377# 0G!H!H)++ '' ' ' '0100Cs## #"1% '  # #A & & &    YY              r)rr)rr )rr)rrrra)# __future__rr)r&platformrrrrrr r typingr namerdsizeofremachinelowerrrr.rkrgrjrqrrrrs StructurerPr`rrrRrrrs0#""""" TTTTTTTTTTTTTTTTTTgo FM&/ * *a / "("$$**,,1 E E E E4        $   6# G=G=G=G=G=G=G=G=T8* * * * * * r