o i'@sddlmZddlZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z  ejdkZeej dkZedvZd!d d Zd"d dZgdZgdZgdZdZdZdZdZGdddejZGdddZ d#ddZ!d$dd Z"dS)%) annotationsN) CFUNCTYPEPOINTERc_longc_size_tc_uint32c_ulongc_void_p)ClassVarnt>amd64x64x86_64x86Zi368zi686-64i686returnstrcCsddlm}|jSNr)Z polars._plrZ_plrZ RUNTIME_REPR)ZplrrH/home/app/Keep/.python/lib/python3.10/site-packages/polars/_cpu_check.pyget_runtime_repr5s r ctypes.CDLLcCsbztjdkrtjdddWStjdddWSty0ddlm}tj|dddYSw) NDarwinz libc.dylibT)Z use_errnoz libc.so.6r)utilc)osunamesysnamectypesCDLL ExceptionrZ find_library)Zctutilrrr_open_posix_libc;s  r")Sr$r$r$_r$Or r$W [)r#r$Ir$Dr$r'r(Ar$r6r$Yr+r6r$r2r r6r$Qr.r/r0)r#r-|$r.r:r4r<r:Lr<r'r(r$r)r$r*r+r$r,r r$r-r.r*r/r0 i@c@s$eZdZUdddDZded<dS) CPUID_structcCsg|]}|tfqSr)r).0rrrr szCPUID_struct.)eaxebxecxZedxz ClassVar[list[tuple[str, type]]]_fields_N)__name__ __module__ __qualname__rJ__annotations__rrrrrCs rCc@s,eZdZdddZddd d Zdd d ZdS)CPUIDrNonec Csjtrtr td|_t}ntjj|_t}ntrt nt}t |}tj ||}trXt |jj _tj tjtjtjg|jj _|j d|ttBt|_|jsOd}t|t|j||nLddl}t}|j}tj tjtjf|_tj|_|d||j|jB|j|jB|_t tj !|j|_|j"|||j||j|j#Bdkrd}t$|t%dt&t't(t(}||j|_)dS)Nz kernel32.dllz)could not allocate memory for CPUID checkrz*could not execute mprotect for CPUID check)* _IS_WINDOWS _IS_64BITrr win_WINDOWS_64_OPCZwindllZkernel32 _CDECL_32_OPC _POSIX_64_OPClenZc_ubyter Z VirtualAllocrestyperrargtypes _MEM_COMMIT _MEM_RESERVE_PAGE_EXECUTE_READWRITEaddr MemoryErrorZmemmovemmapr"mprotectc_int MAP_PRIVATE MAP_ANONYMOUS PROT_READ PROT_WRITEZ addressofZ from_bufferwrite PROT_EXEC RuntimeErrorrrrCrfunc_ptr) selfZopcsizecodemsgr`libcra func_typerrr__init__sR         zCPUID.__init__rrGintrIrCcCst}|||||S)N)rCrj)rkrGrIstructrrr__call__szCPUID.__call__cCs8trt|jj_tttg|jj_|j|j dt dSdSr) rRrrTZ VirtualFreerYr rrrZr^ _MEM_RELEASE)rkrrr__del__s  z CPUID.__del__N)rrP)r)rGrrrIrrrrC)rKrLrMrqrtrvrrrrrOs  =rOdict[str, bool]cCstsiSt}|dd}|dd}|dd}t|jd@t|jd@t|jd@t|jd@t|jd@t|jd @t|jd @t|jd @t|jd @t|jd @t|jd@t|jd@t|jd@t|jd@dS)Nr7rr)lir@rAiii@iir  )Zsse3Zssse3fmaZ cmpxchg16bzsse4.1zsse4.2ZmovbeZpopcntZ pclmulqdqZavxZbmi1Zbmi2Zavx2Zlzcnt)_SUPPORTS_CPUIDrOboolrIrH)ZcpuidZcpuid1Zcpuid7Zcpuid81hrrr_read_cpu_flagss*                 r~ feature_flagsrPcCsdd|dD}dd|D}|rtjdrdSt}g}|D]}||vr1d|}t|||s:||q"|rRddl}|jdd |d t d d dSdS) NcSs g|] }|ds|dqS)-+) startswithlstriprDfrrrrFs  z#check_cpu_flags..,cSsg|] }|r |dkr|qS)z ctr-staticrrrrrrFsZPOLARS_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. r7) stacklevel) splitrenvirongetr~riappendwarningswarnjoinRuntimeWarning)rZexpected_cpu_flagsZsupported_cpu_flagsZmissing_featuresrrnrrrrcheck_cpu_flagss6    r)rr)rr)rrw)rrrrP)# __future__rrrplatformrrrrrrr typingr namerRZsizeofrSmachinelowerr|rr"rWrUrVr[r\rur]Z StructurerCrOr~rrrrrs. $     J