diff --git a/faiss/python/setup.py b/faiss/python/setup.py index 8d2150e7f7..ea623ee1b2 100644 --- a/faiss/python/setup.py +++ b/faiss/python/setup.py @@ -26,11 +26,13 @@ swigfaiss_generic_lib = f"{prefix}_swigfaiss{ext}" swigfaiss_avx2_lib = f"{prefix}_swigfaiss_avx2{ext}" swigfaiss_avx512_lib = f"{prefix}_swigfaiss_avx512{ext}" +callbacks_lib = f"{prefix}libfaiss_python_callbacks{ext}" swigfaiss_sve_lib = f"{prefix}_swigfaiss_sve{ext}" found_swigfaiss_generic = os.path.exists(swigfaiss_generic_lib) found_swigfaiss_avx2 = os.path.exists(swigfaiss_avx2_lib) found_swigfaiss_avx512 = os.path.exists(swigfaiss_avx512_lib) +found_callbacks = os.path.exists(callbacks_lib) found_swigfaiss_sve = os.path.exists(swigfaiss_sve_lib) assert (found_swigfaiss_generic or found_swigfaiss_avx2 or found_swigfaiss_avx512 or found_swigfaiss_sve), \ @@ -53,6 +55,10 @@ shutil.copyfile("swigfaiss_avx512.py", "faiss/swigfaiss_avx512.py") shutil.copyfile(swigfaiss_avx512_lib, f"faiss/_swigfaiss_avx512{ext}") +if found_callbacks: + print(f"Copying {callbacks_lib}") + shutil.copyfile(callbacks_lib, f"faiss/{callbacks_lib}") + if found_swigfaiss_sve: print(f"Copying {swigfaiss_sve_lib}") shutil.copyfile("swigfaiss_sve.py", "faiss/swigfaiss_sve.py")