jP ddlZddlZddlZddlZdZe[ddlddlmZddl ddl ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!de"e#e$fzZ%ejLe'Z(ejRe*ejVe,ejZe.ej^e0ejbe2ejfe4ejje6ejne8ejre:ejve<ejze>d ejze?d ejze@d d ejeBejeDejeEejeGejeIejeKejeMejeOZOeje'ZQeReQD]ZSeTeQeSZUejeUseUZWeXeWeYrejeWeXeWe[rejeWeXeWe]rejeWeXeWe_rejeWeXeWeas eXeWebrejeWeXeWedrejeWeXeWefsejeWd Zhd ZidZjdZkdZldemvr-ehendeheodehepdeheqdeherdehesdehetdeheudehevddgdgdejevddehewdehexdeheydehezdehe{dehe|dehe}dehe~deke~dehedehedehedeheadehebdehedehedehedejeddejeddehedgdgdehedddgiehedddgiehedehedehedehedejeddejeddehedehedehedeheedeheedeheedeheedeheedeheedehedehedeheddemvr$ehedehedehedehedeZd#dZeZ d#dZeZdZeZeZeZeZd$dZd$dZdZd ZGd!d"Z ey#e$rYywxYw)%Ncd ddlm}ddlddlfd} ddlm}d}|rc ddlm}jjt|jdd }d D]$}|jj||&nfd }|d d} |dd} |dd} |jj| d|jj| d|jj| d|jj| dy ddl } ddl} ddl}|| | fD]}|j!y#t$rYywxYw#t$rd}YSwxYw#t$r}td |d}~wwxYw#t$r}td|d}~wwxYw)aPre-load CUDA / RAPIDS shared libs from pip wheels with RTLD_GLOBAL. These libs ship in nvidia-*-cuNN / libcuvs-cuNN wheels, off ld.so's search path, so we dlopen them before the SWIG extension loads libfaiss.so. Gated on the `faiss._gpu_build` marker (CMake writes it only for GPU builds); the `_cuvs_build` marker selects the CUDA 13 cuVS variant (else CUDA 12) and adds the cuVS stack. Missing wheels raise a fix-it. ) _gpu_buildNrc j|jy#t$r1}tdjj |d|d|d}~wwxYw)z9dlopen one .so with global visibility, or raise a fix-it.)modezfaiss-gpu: failed to load z from u- — corrupt or incomplete nvidia CUDA wheel?N)CDLL RTLD_GLOBALOSError RuntimeErrorpathbasename)r ectypesoss X/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/faiss/__init__.py_loadz _preload_gpu_libs.._load"sg  KK6#5#5K 6 ,RWW-=-=d-C,DF&EG  s! A,AA) _cuvs_buildTFlibufaiss-gpu-cuvs installed but the CUDA 13 runtime wheels are missing — pip install 'nvidia-cuda-runtime>=13.2,<14' 'nvidia-cublas>=13.2,<14' 'nvidia-curand>=10.3.7,<11' 'nvidia-nvjitlink>=13.2,<14')zlibcudart.so.13zlibnvJitLink.so.13zlibcublas.so.13zlibcublasLt.so.13libcurand.so.10c2 td|z|g}jj |j dd S#t$rP}|jddjddjdd}td|d|d |d }~wwxYw) zMReturn the lib/ dir of an nvidia-*-cu12 wheel, or raise a fix-it.znvidia.)fromlist>r<=zfaiss-gpu installed but u is missing — pip install ''Nr) __import__ ImportErrorsplitr r join__path__) import_namepip_specmodrpip_namers r_nvidia_lib_dirz*_preload_gpu_libs.._nvidia_lib_dirTs   +{m77<< Q7 7 #>>#.q177=12.6,<13cublasznvidia-cublas-cu12>=12.6,<13curandznvidia-curand-cu12>=10.3.7,<11zlibcudart.so.12zlibcublas.so.12zlibcublasLt.so.12rufaiss-gpu-cuvs installed but the cuVS runtime wheels are missing — pip install 'libcuvs-cu13>=26.06,<27' --extra-index-url https://pypi.nvidia.com)rrrrr nvidia.cu13cu13r rlistr r libcuvslibraftlibrmm load_library)rrris_cuvs_cu13_libdirr_sonamer%_cudart_cublas_curandr-r.r/_modrrs @@r_preload_gpu_libsr9s  !  'ggll4#7#:EBG G "'',,w0 1  8"" @ "(,JK!(,LM bggll7$567 bggll7$567 bggll7$789 bggll7$567 '* +I 2  /    n  8   sQEE&8E80 F E#"E#& E54E58 F F  F F/ F**F/r)*)class_wrappers)kminkmaxpairwise_distancesrandrandintlrandrandnrand_smooth_vectorseval_intersection normalize_L2 ResultHeapknnKmeans SuperKmeanschecksummatrix_bucket_sort_inplace bucket_sortmerge_knn_resultsMapInt64ToInt64 knn_hammingpack_bitstringsunpack_bitstringsz%d.%d.%dT) class_ownsF)rR force_int64cz|jfd}fd}ttk(r||_y||_y)Nc.|g||g|_yNreferenced_objects)selfargs original_init parameter_nos rreplacement_initz0add_ref_in_constructor..replacement_inits d"T"#' #5"6cn|g|t|}|Dcgc]}|| c}|_ycc}wrV)lenrX)rYrZpsetnor[r\s rreplacement_init_multiplez9add_ref_in_constructor..replacement_init_multiples<d"T"CI&6:";d48d";";s 2)__init__typedict) the_classr\r]rcr[s ` @radd_ref_in_constructorrhs9&&M7<  LT!6 - r^cdt|ds |g|_y|jj|yNrX)hasattrrXappend)rYrefs radd_to_referenced_objectsrn s+ 4- .#&% &&s+r^cFt||fd}t|||y)Nc8|}t|||g|SrV)rn)rYrZrmoriginal_methodr\s rreplacement_methodz-add_ref_in_method..replacement_methods'< !$,t+d++r^getattrsetattr)rg method_namer\rrrqs ` @radd_ref_in_methodrws#i5O,  I{$67r^cDt||dfd }t|||y)Nc|s1t|ds |g|_n6|jj|n|jj |||Srj)rkrXrlthisdisown)rYrmownrqs rrrz:add_ref_in_method_explicit_own..replacement_method"sN4!56+.%'''..s3 HHOO tS#..r^)Frs)rgrvrrrqs @radd_ref_in_method_explicit_ownr}s#i5O / I{$67r^cVtt|fd}tt||y)Nc,|}|}|g|_|SrVrW)rZresultrmoriginal_functionr\s rreplacement_functionz1add_ref_in_function..replacement_function4s&"D)< %(E! r^)rt this_moduleru) function_namer\rrs ` @radd_ref_in_functionr0s$ ];  K(<=r^GPU)rprepend_transform replace_vt add_shardraddIndexSVSc tj|d}|j\}}||jk(sJ|s7t }t |}|j |_|j|_tjdd}tjdd} tj||ftj} tj||ftj} t||t||t| t| |t|t|  |s| | fS|d| d| d| dd } | | | fS) Nfloat32dtyperuint64rfloat64rrndispre_transform_mscoarse_quantizer_msinvlist_scan_ms) npascontiguousarrayshapedIVFSearchParametersextract_index_ivfnprobe max_codesemptyrint64search_with_parameters_cswig_ptr) indexxkparams output_statsnr index_ivfnb_dis ms_per_stage distanceslabelsstatss rsearch_with_parametersrs0 Qi0A 77DAq << $&%e, !(( $.. XXa "F88Ay)L!Qrzz2I XXq!fBHH -F      &  1I ,Q#/?+A   &%''r^c tj|d}|j\}}||jk(sJ|s7t }t |}|j |_|j|_tjdd}tjdd} t|} t||t||| |t|t| t| j|dzj} t| d} t| j | j} t| j"| j}|s| | |fS|d| d| d| d d }| | ||fS) Nrrrrrrrrr)rrrrrrrrrRangeSearchResultrange_search_with_parameters_cr rev_swig_ptrlimscopyintrr)rrradiusrrrrrrrresrndDoutIoutrs rrange_search_with_parametersrs_ Qi0A 77DAq << $&%e, !(( $.. XXa "F88Ay)L A C"      !a% ( - - /D T"XB  r * / / 1D  B ' , , .D T41I ,Q#/?+A   T4&&r^c\d|dfd|dfd|dfd|dfd|dffD]Q\}}} |j| k7rtd|d | d |j|jd rDtd|d |jd k7rtd|j |j \} } |jd k7s|j d| k7rtd| d|j |j d} |j | fk7rtd| d|j |j | fk7rtd| d|j |j | dzfk7rtd| dzd|j t jdt j} t jdt j}tt|| | t|| t|t||t||t| t| t| dt|dfS)aRun one SuperKMeans iter-1+ pruned assignment pass on caller-managed state. All arrays must be C-contiguous. `X_tilde`, `Y_tilde`, `tau`, and `ad_coeff` must be float32; `assignments` must be int32. These mirror the C++ pointer contract, and passing another dtype (e.g. int64 assignments, numpy's default integer type) would otherwise reinterpret the buffer and corrupt results. Shapes: `X_tilde` is (n, d), `Y_tilde` is (k, d), `tau` and `assignments` have length n, and `ad_coeff` has length d + 1. Returns (total_pairs, pruned_at_gemm) for a d_prime controller. Mutates `tau` and `assignments` in place. X_tilderY_tildetauad_coeff assignmentsint32zsuper_kmeans_assign_iteration: z must be z, got C_CONTIGUOUSz must be C-contiguousrzDsuper_kmeans_assign_iteration: X_tilde must be 2D (n, d), got shape rz;super_kmeans_assign_iteration: Y_tilde must have shape (k, z), got rz4super_kmeans_assign_iteration: tau must have shape (z,), got z>" $   HHQbhh 'E XXarxx (F#1a1 x ,(#R&)  uQx=#fQi. ((r^cZt}t|||t|jSz'convert an index to a numpy uint8 array)VectorIOWriter write_indexvector_to_arraydata)rio_flagswriters rserialize_indexr2s%  Fvx( 6;; ''r^cZt}t||jt||SrV)VectorIOReadercopy_array_to_vectorr read_index)rrreaders rdeserialize_indexr9s%  Fv{{+ fh ''r^cXt}t||t|jSr)rwrite_index_binaryrr)rrs rserialize_index_binaryr?s#  Fuf% 6;; ''r^cXt}t||jt|SrV)rrrread_index_binary)rrs rdeserialize_index_binaryrFs#  Fv{{+ V $$r^c$eZdZdefdZdZdZy) TimeoutGuardtimeout_in_secondsc||_yrV)timeout)rYrs rrdzTimeoutGuard.__init__Ms ) r^cBtj|jyrV)TimeoutCallbackresetr)rYs r __enter__zTimeoutGuard.__enter__Psdll+r^c,tjyrV)PythonInterruptCallbackr)rYexc_type exc_value tracebacks r__exit__zTimeoutGuard.__exit__Ss%%'r^N)__name__ __module__ __qualname__floatrdrrr^rrrLs*5*,(r^r)NF)r)numpyrloggingsysinspectr9loaderfaissr;faiss.gpu_wrappersfaiss.array_conversionsfaiss.extra_wrappersr<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQFAISS_VERSION_MAJORFAISS_VERSION_MINORFAISS_VERSION_PATCH __version__ getLoggerrloggerhandle_Clustering Clusteringhandle_SuperKMeans SuperKMeanshandle_Clustering1D Clustering1Dhandle_MatrixStats MatrixStatshandle_IOWriterIOWriterhandle_IOReaderIOReaderhandle_AutoTuneCriterionAutoTuneCriterionhandle_ParameterSpaceParameterSpace handle_NSGIndexNSGhandle_MapLong2Long MapLong2Longhandle_IDSelectorSubsetIDSelectorBatchIDSelectorArrayIDSelectorBitmaphandle_CodeSetCodeSethandle_Tensor2DTensor2D Int32Tensor2Dhandle_Embedding Embedding handle_LinearLinear handle_QINCoQINCohandle_QINCoStep QINCoStep handle_shard_ivf_index_centroidsshard_ivf_index_centroidsmodulesrdirsymbolrtobjisclassrg issubclassIndex handle_Index IndexBinaryhandle_IndexBinaryVectorTransformhandle_VectorTransform Quantizerhandle_QuantizerIndexRowwiseMinMaxIndexRowwiseMinMaxFP16handle_IndexRowwiseMinMaxSearchParametershandle_SearchParameters CodePackerhandle_CodePackerrhrnrwr}rget_compile_optionsGpuIndexIVFFlatGpuIndexBinaryFlat GpuIndexFlat GpuIndexIVFPQGpuIndexIVFScalarQuantizer IndexIVFFlatIndexIVFFlatDedupIndexIVFFlatPanoramaIndexPreTransform IndexIVFPQ IndexIVFPQRIndexIVFPQFastScanIndexIVFResidualQuantizerIndexIVFLocalSearchQuantizer!IndexIVFResidualQuantizerFastScan$IndexIVFLocalSearchQuantizerFastScanIndexIVFSpectralHash Index2LayerLevel1QuantizerIndexIVFScalarQuantizer IndexIDMap IndexIDMap2 IndexHNSW IndexShardsIndexBinaryShardsIndexRefineFlatIndexRefinePanorama IndexRefineIndexBinaryIVFIndexBinaryFromFloatIndexBinaryIDMapIndexBinaryIDMap2 IndexReplicasIndexBinaryReplicasBufferedIOWriterBufferedIOReader IDSelectorNot IDSelectorAndslice IDSelectorOr IDSelectorXOrIDSelectorTranslatedIndexIVFIndependentQuantizerIndexIVFRaBitQIndexIVFRaBitQFastScan IndexIVFEDENIndexSVSVamanaIndexSVSVamanaLVQIndexSVSVamanaLeanVec IndexSVSFlatrrrrrr IndexProxyHStackInvertedListsConcatenatedInvertedListsIndexResidualQuantizer IndexResidualSearchParametersIVFrrrrrrpost_init_hook NameErrorrr^rrs$ pf! %4   8 $  ,!!!+."""<0!!!+.x(x('''(9:$$$^4(#"""<0&&&4H&&&5I&&&Eg&x(}- *V$E" *KNKK kk(# +F +v &Cws i ' 'N ' ' 2 i - -N - -i 8 i 1 1N 1 1) < i + +N + +I 6 i!3 4 -9  5N 4 4Y ? i!1 2 2N 2 29 = i , ,N , ,Y 75P.*,88$ >  !!?A.-q1<+=!,5q9|Q'(!,+Q/(q!f!*=>#%8!<z1%{A&)1-0!43Q78!<;Q?+Q/3\B{A&*.2)1--q1z1%{A&y!$+{A.#[!4QCQC(89*QAK8{QAK0~q)+Q/'+(!,-Q/%z15'+'+}a(}eAh/|U1X.}eAh/+U1X6}eAh/3U1X>~q)-q1|Q'  !!>1-,a00!4<+2"(J">16&'R#@D)R /& )(( (% ((   s3U;;VV