o gS @sddlZddlZddlZddlZddZddZeeZdZ dZ ej e dZ e dur9ede deZ ned e d eZ e e d Zed d e DZerzedddlTeddZWneyZzeded ZWYdZ[ndZ[wwedd e DZerzedddlTeddZWneyZzeded ZWYdZ[ndZ[wwde vZereszedddlTeddZWneyZzeded ZWYdZ[ndZ[wwde vZer1es1zedddlTeddZWney0Zzeded ZWYdZ[ndZ[wwesDed ddlTed!dSdS)"NcCsdd|dDS)NcSsg|]}t|qS)int.0xrrf/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/faiss/loader.py szVersion...)split)vrrrVersion sr cCsldd}ddl}t|jtdkr;ddlm}dd|D}|r(|d td d  d D]}| |q1|St d krRt ddgddkrOdhStSt dkrddl}t}d|jjjjddd vrr|dd|jjjjddd vr|dd|jjjjddd vr|d|r|d td d  d D]}| |q|StS)a Returns the set of supported CPU features, see https://github.com/numpy/numpy/blob/master/numpy/core/src/common/npy_cpu_features.h for the list of features that this set may contain per architecture. Example: >>> supported_instruction_sets() # for x86 {"SSE2", "AVX2", "AVX512", ...} >>> supported_instruction_sets() # for PPC {"VSX", "VSX2", ...} >>> supported_instruction_sets() # for ARM {"NEON", "ASIMD", ...} cSsftdkrdStdkrdSddl}t|jtdkrdSddl}d|jjj j d dd vS) Naarch64FLinuxrz2.0sveFeatures) platformmachinesystemnumpyr __version__numpy.distutils.cpuinfo distutilscpuinfocpuinfogetr )rrrris_sve_supported s   z4supported_instruction_sets..is_sve_supportedrNz1.19)__cpu_features__cSsh|]\}}|r|qSrr)rkr rrr 4sz-supported_instruction_sets..SVEFAISS_DISABLE_CPU_FEATURESrz, Darwinz/usr/sbin/sysctlzhw.optional.avx2_01AVX2ravx2flagsavx512AVX512 avx512_fp16 AVX512_SPR)rr rnumpy.core._multiarray_umathritemsaddosgetenvr discardrr subprocess check_outputrsetrrrrr)rrr supportedfresultrrrsupported_instruction_setss:          r9FAISS_OPT_LEVELzEnvironment variable zK is not set, so let's pick the instruction set according to the current CPUzUsing z as an instruction set.Fcc|] }d|vVqdS)r,Nupperrrrr br>z&Loading faiss with AVX512-SPR support.)*z2Successfully loaded faiss with AVX512-SPR support.Tz7Could not load library with AVX512-SPR support due to: ccr;)r*Nr<rrrrr>nr?z"Loading faiss with AVX512 support.z.Successfully loaded faiss with AVX512 support.z3Could not load library with AVX512 support due to: r&z Loading faiss with AVX2 support.z,Successfully loaded faiss with AVX2 support.z1Could not load library with AVX2 support due to: r!zLoading faiss with SVE support.z+Successfully loaded faiss with SVE support.z0Could not load library with SVE support due to: zLoading faiss.zSuccessfully loaded faiss.)rr3loggingr0r r9 getLogger__name__loggerinstruction_setsopt_env_variable_nameenvironr opt_leveldebugr5r/loadedanyhas_AVX512_SPRrswigfaiss_avx512_spr ImportErrore has_AVX512swigfaiss_avx512has_AVX2swigfaiss_avx2has_SVE swigfaiss_sve swigfaissrrrrs A