Initial commit
This commit is contained in:
33
packages/leann-backend-hnsw/third_party/faiss/.github/ISSUE_TEMPLATE.md
vendored
Normal file
33
packages/leann-backend-hnsw/third_party/faiss/.github/ISSUE_TEMPLATE.md
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
# Summary
|
||||
|
||||
<!-- Facebook has a bounty program for the safe disclosure of security bugs. In
|
||||
those cases, please go through the process outlined on that page and do not
|
||||
file a public issue. -->
|
||||
|
||||
# Platform
|
||||
|
||||
<!-- if the question/problem is not platform-specific, please ignore this -->
|
||||
|
||||
OS: <!-- e.g. macOS 10.13.3 -->
|
||||
|
||||
Faiss version: <!-- git commit, e.g. 56383610bcb982d6591e2e2bea3516cb7723e04a -->
|
||||
|
||||
Installed from: <!-- anaconda? compiled by yourself ? -->
|
||||
|
||||
Faiss compilation options: <!-- e.g. using MKL with compile flags ... -->
|
||||
|
||||
Running on:
|
||||
- [ ] CPU
|
||||
- [ ] GPU
|
||||
|
||||
Interface:
|
||||
- [ ] C++
|
||||
- [ ] Python
|
||||
|
||||
# Reproduction instructions
|
||||
|
||||
<!-- Please provide specific and comprehensive instructions to reproduce the
|
||||
described behavior. -->
|
||||
|
||||
<!-- Please *do not* post screenshots of logs. They are not searchable. Copy/paste
|
||||
the text or make a gist if the text is too bulky. -->
|
||||
189
packages/leann-backend-hnsw/third_party/faiss/.github/actions/build_cmake/action.yml
vendored
Normal file
189
packages/leann-backend-hnsw/third_party/faiss/.github/actions/build_cmake/action.yml
vendored
Normal file
@@ -0,0 +1,189 @@
|
||||
name: Build cmake
|
||||
inputs:
|
||||
opt_level:
|
||||
description: 'Compile options / optimization level.'
|
||||
required: false
|
||||
default: generic
|
||||
gpu:
|
||||
description: 'Enable GPU support.'
|
||||
required: false
|
||||
default: OFF
|
||||
cuvs:
|
||||
description: 'Enable cuVS support.'
|
||||
required: false
|
||||
default: OFF
|
||||
rocm:
|
||||
description: 'Enable ROCm support.'
|
||||
required: false
|
||||
default: OFF
|
||||
runs:
|
||||
using: composite
|
||||
steps:
|
||||
- name: Setup miniconda
|
||||
uses: conda-incubator/setup-miniconda@v3
|
||||
with:
|
||||
python-version: '3.11'
|
||||
miniforge-version: latest # ensures conda-forge channel is used.
|
||||
channels: conda-forge
|
||||
conda-remove-defaults: 'true'
|
||||
# Set to aarch64 if we're on arm64 because there's no miniforge ARM64 package, just aarch64.
|
||||
# They are the same thing, just named differently.
|
||||
architecture: ${{ runner.arch == 'ARM64' && 'aarch64' || runner.arch }}
|
||||
- name: Configure build environment
|
||||
shell: bash
|
||||
run: |
|
||||
# initialize Conda
|
||||
conda config --set solver libmamba
|
||||
# Ensure starting packages are from conda-forge.
|
||||
conda list --show-channel-urls
|
||||
conda update -y -q conda
|
||||
echo "$CONDA/bin" >> $GITHUB_PATH
|
||||
|
||||
conda install -y -q python=3.11 cmake=3.26 make=4.2 swig=4.0 "numpy<2" scipy=1.14 pytest=7.4 gflags=2.2
|
||||
|
||||
# install base packages for ARM64
|
||||
if [ "${{ runner.arch }}" = "ARM64" ]; then
|
||||
conda install -y -q -c conda-forge openblas=0.3.29 gxx_linux-aarch64=14.2 sysroot_linux-aarch64=2.17
|
||||
fi
|
||||
|
||||
# install base packages for X86_64
|
||||
if [ "${{ runner.arch }}" = "X64" ]; then
|
||||
# TODO: merge this with ARM64
|
||||
conda install -y -q -c conda-forge gxx_linux-64=14.2 sysroot_linux-64=2.17
|
||||
conda install -y -q mkl=2022.2.1 mkl-devel=2022.2.1
|
||||
fi
|
||||
|
||||
# no CUDA needed for ROCm so skip this
|
||||
if [ "${{ inputs.rocm }}" = "ON" ]; then
|
||||
:
|
||||
# regular CUDA for GPU builds
|
||||
elif [ "${{ inputs.gpu }}" = "ON" ] && [ "${{ inputs.cuvs }}" = "OFF" ]; then
|
||||
conda install -y -q cuda-toolkit=12.4 -c "nvidia/label/cuda-12.4.0"
|
||||
# and CUDA from cuVS channel for cuVS builds
|
||||
elif [ "${{ inputs.cuvs }}" = "ON" ]; then
|
||||
conda install -y -q libcuvs=24.12 'cuda-version>=12.0,<=12.5' cuda-toolkit=12.4.1 gxx_linux-64=12.4 -c rapidsai -c conda-forge
|
||||
fi
|
||||
|
||||
# install test packages
|
||||
if [ "${{ inputs.rocm }}" = "ON" ]; then
|
||||
: # skip torch install via conda, we need to install via pip to get
|
||||
# ROCm-enabled version until it's supported in conda by PyTorch
|
||||
elif [ "${{ inputs.gpu }}" = "ON" ]; then
|
||||
conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c pytorch -c "nvidia/label/cuda-12.4.0"
|
||||
else
|
||||
conda install -y -q "pytorch<2.5" -c pytorch
|
||||
fi
|
||||
- name: ROCm - Install dependencies
|
||||
if: inputs.rocm == 'ON'
|
||||
shell: bash
|
||||
run: |
|
||||
# Update repos and install kmod, wget, gpg
|
||||
sudo apt-get -qq update >/dev/null
|
||||
sudo apt-get -qq install -y kmod wget gpg >/dev/null
|
||||
|
||||
# Get UBUNTU version name
|
||||
UBUNTU_VERSION_NAME=`cat /etc/os-release | grep UBUNTU_CODENAME | awk -F= '{print $2}'`
|
||||
|
||||
# Set ROCm version
|
||||
ROCM_VERSION="6.2"
|
||||
|
||||
# Download, prepare, and install the package signing key
|
||||
mkdir --parents --mode=0755 /etc/apt/keyrings
|
||||
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | gpg --dearmor | sudo tee /etc/apt/keyrings/rocm.gpg > /dev/null
|
||||
|
||||
# Add rocm repository
|
||||
wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
|
||||
rocm_baseurl="http://repo.radeon.com/rocm/apt/${ROCM_VERSION}"
|
||||
echo "deb [arch=amd64] ${rocm_baseurl} ${UBUNTU_VERSION_NAME} main" | sudo tee /etc/apt/sources.list.d/rocm.list
|
||||
sudo apt-get -qq update --allow-insecure-repositories >/dev/null
|
||||
sudo apt-get -qq install -y --allow-unauthenticated \
|
||||
"rocm-dev${ROCM_VERSION}" "rocm-utils${ROCM_VERSION}" \
|
||||
"rocm-libs${ROCM_VERSION}" >/dev/null
|
||||
|
||||
# Fake presence of MI200-class accelerators
|
||||
echo "gfx90a" | sudo tee /opt/rocm/bin/target.lst
|
||||
|
||||
# Cleanup
|
||||
sudo apt-get -qq autoclean >/dev/null
|
||||
sudo apt-get -qq clean >/dev/null
|
||||
sudo rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
|
||||
- name: Symblink system dependencies
|
||||
if: inputs.rocm == 'ON'
|
||||
shell: bash
|
||||
run: |
|
||||
# symblink system libraries for HIP compiler
|
||||
sudo ln -s /lib/x86_64-linux-gnu/libc.so.6 /lib64/libc.so.6
|
||||
sudo ln -s /lib/x86_64-linux-gnu/libc_nonshared.a /usr/lib64/libc_nonshared.a
|
||||
sudo ln -s /usr/lib/x86_64-linux-gnu/libpthread.so.0 /lib64/libpthread.so.0
|
||||
sudo ln -s $HOME/miniconda3/x86_64-conda-linux-gnu/sysroot/usr/lib64/libpthread_nonshared.a /usr/lib64/libpthread_nonshared.a
|
||||
- name: Build all targets
|
||||
shell: bash
|
||||
run: |
|
||||
eval "$(conda shell.bash hook)"
|
||||
conda activate
|
||||
cmake -B build \
|
||||
-DBUILD_TESTING=ON \
|
||||
-DBUILD_SHARED_LIBS=ON \
|
||||
-DFAISS_ENABLE_GPU=${{ inputs.gpu }} \
|
||||
-DFAISS_ENABLE_CUVS=${{ inputs.cuvs }} \
|
||||
-DFAISS_ENABLE_ROCM=${{ inputs.rocm }} \
|
||||
-DFAISS_OPT_LEVEL=${{ inputs.opt_level }} \
|
||||
-DFAISS_ENABLE_C_API=ON \
|
||||
-DPYTHON_EXECUTABLE=$CONDA/bin/python \
|
||||
-DCMAKE_BUILD_TYPE=Release \
|
||||
-DBLA_VENDOR=${{ runner.arch == 'X64' && 'Intel10_64_dyn' || '' }} \
|
||||
-DCMAKE_CUDA_FLAGS=${{ runner.arch == 'X64' && '"-gencode arch=compute_75,code=sm_75"' || '' }} \
|
||||
.
|
||||
make -k -C build -j$(nproc)
|
||||
- name: C++ tests
|
||||
shell: bash
|
||||
run: |
|
||||
export GTEST_OUTPUT="xml:$(realpath .)/test-results/googletest/"
|
||||
make -C build test
|
||||
- name: C++ perf benchmarks
|
||||
shell: bash
|
||||
if: inputs.rocm == 'OFF'
|
||||
run: |
|
||||
find ./build/perf_tests/ -executable -type f -name "bench*" -exec '{}' -v \;
|
||||
- name: Install Python extension
|
||||
shell: bash
|
||||
working-directory: build/faiss/python
|
||||
run: |
|
||||
$CONDA/bin/python setup.py install
|
||||
- name: ROCm - install ROCm-enabled torch via pip
|
||||
if: inputs.rocm == 'ON'
|
||||
shell: bash
|
||||
run: |
|
||||
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1
|
||||
- name: Python tests (CPU only)
|
||||
if: inputs.gpu == 'OFF'
|
||||
shell: bash
|
||||
run: |
|
||||
pytest --junitxml=test-results/pytest/results.xml tests/test_*.py
|
||||
pytest --junitxml=test-results/pytest/results-torch.xml tests/torch_*.py
|
||||
- name: Python tests (CPU + GPU)
|
||||
if: inputs.gpu == 'ON'
|
||||
shell: bash
|
||||
run: |
|
||||
pytest --junitxml=test-results/pytest/results.xml tests/test_*.py
|
||||
pytest --junitxml=test-results/pytest/results-torch.xml tests/torch_*.py
|
||||
cp tests/common_faiss_tests.py faiss/gpu/test
|
||||
pytest --junitxml=test-results/pytest/results-gpu.xml faiss/gpu/test/test_*.py
|
||||
pytest --junitxml=test-results/pytest/results-gpu-torch.xml faiss/gpu/test/torch_*.py
|
||||
- name: Test avx2 loading
|
||||
if: inputs.opt_level == 'avx2'
|
||||
shell: bash
|
||||
run: |
|
||||
FAISS_DISABLE_CPU_FEATURES=AVX2 LD_DEBUG=libs $CONDA/bin/python -c "import faiss" 2>&1 | grep faiss.so
|
||||
LD_DEBUG=libs $CONDA/bin/python -c "import faiss" 2>&1 | grep faiss_avx2.so
|
||||
- name: Upload test results
|
||||
if: always()
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: test-results-arch=${{ runner.arch }}-opt=${{ inputs.opt_level }}-gpu=${{ inputs.gpu }}-cuvs=${{ inputs.cuvs }}-rocm=${{ inputs.rocm }}
|
||||
path: test-results
|
||||
- name: Check installed packages channel
|
||||
shell: bash
|
||||
run: |
|
||||
# Shows that all installed packages are from conda-forge.
|
||||
conda list --show-channel-urls
|
||||
107
packages/leann-backend-hnsw/third_party/faiss/.github/actions/build_conda/action.yml
vendored
Normal file
107
packages/leann-backend-hnsw/third_party/faiss/.github/actions/build_conda/action.yml
vendored
Normal file
@@ -0,0 +1,107 @@
|
||||
name: Conda build
|
||||
description: Builds Faiss inside a Conda environment and uploads to repository when label is provided.
|
||||
inputs:
|
||||
label:
|
||||
description: "The label to be used for uploads to Conda."
|
||||
default: ""
|
||||
required: false
|
||||
cuda:
|
||||
description: "CUDA toolkit version to use."
|
||||
default: ""
|
||||
required: false
|
||||
cuvs:
|
||||
description: "Enable cuVS support."
|
||||
default: ""
|
||||
required: false
|
||||
runs:
|
||||
using: composite
|
||||
steps:
|
||||
- name: Choose shell
|
||||
shell: bash
|
||||
id: choose_shell
|
||||
run: |
|
||||
# Use pwsh on Windows; bash everywhere else
|
||||
if [ "${{ runner.os }}" != "Windows" ]; then
|
||||
echo "shell=bash" >> "$GITHUB_OUTPUT"
|
||||
else
|
||||
echo "shell=pwsh" >> "$GITHUB_OUTPUT"
|
||||
fi
|
||||
- name: Setup miniconda
|
||||
uses: conda-incubator/setup-miniconda@v3
|
||||
with:
|
||||
python-version: '3.11'
|
||||
miniforge-version: latest # ensures conda-forge channel is used.
|
||||
channels: conda-forge
|
||||
conda-remove-defaults: 'true'
|
||||
# Set to runner.arch=aarch64 if we're on arm64 because
|
||||
# there's no miniforge ARM64 package, just aarch64.
|
||||
# They are the same thing, just named differently.
|
||||
# However there is an ARM64 for macOS, so exclude that.
|
||||
architecture: ${{ (runner.arch == 'ARM64' && runner.os != 'macOS') && 'aarch64' || runner.arch }}
|
||||
- name: Install conda build tools
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
run: |
|
||||
# Ensure starting packages are from conda-forge.
|
||||
conda list --show-channel-urls
|
||||
conda install -y -q "conda!=24.11.0"
|
||||
conda install -y -q "conda-build!=24.11.0" "liblief=0.14.1"
|
||||
conda list --show-channel-urls
|
||||
- name: Enable anaconda uploads
|
||||
if: inputs.label != ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
env:
|
||||
PACKAGE_TYPE: ${{ inputs.label }}
|
||||
run: |
|
||||
conda install -y -q anaconda-client
|
||||
conda config --set anaconda_upload yes
|
||||
- name: Conda build (CPU)
|
||||
if: inputs.label == '' && inputs.cuda == ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
run: |
|
||||
conda build faiss --python 3.11 -c pytorch
|
||||
- name: Conda build (CPU) w/ anaconda upload
|
||||
if: inputs.label != '' && inputs.cuda == ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
env:
|
||||
PACKAGE_TYPE: ${{ inputs.label }}
|
||||
run: |
|
||||
conda build faiss --user pytorch --label ${{ inputs.label }} -c pytorch
|
||||
- name: Conda build (GPU)
|
||||
if: inputs.label == '' && inputs.cuda != '' && inputs.cuvs == ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
run: |
|
||||
conda build faiss-gpu --variants '{ "cudatoolkit": "${{ inputs.cuda }}" }' \
|
||||
-c pytorch -c nvidia/label/cuda-${{ inputs.cuda }} -c nvidia
|
||||
- name: Conda build (GPU) w/ anaconda upload
|
||||
if: inputs.label != '' && inputs.cuda != '' && inputs.cuvs == ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
env:
|
||||
PACKAGE_TYPE: ${{ inputs.label }}
|
||||
run: |
|
||||
conda build faiss-gpu --variants '{ "cudatoolkit": "${{ inputs.cuda }}" }' \
|
||||
--user pytorch --label ${{ inputs.label }} -c pytorch -c nvidia/label/cuda-${{ inputs.cuda }} -c nvidia
|
||||
- name: Conda build (GPU w/ cuVS)
|
||||
if: inputs.label == '' && inputs.cuda != '' && inputs.cuvs != ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
run: |
|
||||
conda build faiss-gpu-cuvs --variants '{ "cudatoolkit": "${{ inputs.cuda }}" }' \
|
||||
-c pytorch -c rapidsai -c rapidsai-nightly -c conda-forge -c nvidia
|
||||
- name: Conda build (GPU w/ cuVS) w/ anaconda upload
|
||||
if: inputs.label != '' && inputs.cuda != '' && inputs.cuvs != ''
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
working-directory: conda
|
||||
env:
|
||||
PACKAGE_TYPE: ${{ inputs.label }}
|
||||
run: |
|
||||
conda build faiss-gpu-cuvs --variants '{ "cudatoolkit": "${{ inputs.cuda }}" }' \
|
||||
--user pytorch --label ${{ inputs.label }} -c pytorch -c rapidsai -c rapidsai-nightly -c conda-forge -c nvidia
|
||||
- name: Check installed packages channel
|
||||
shell: ${{ steps.choose_shell.outputs.shell }}
|
||||
run: |
|
||||
# Shows that all installed packages are from conda-forge.
|
||||
conda list --show-channel-urls
|
||||
23
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/autoclose.yml
vendored
Normal file
23
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/autoclose.yml
vendored
Normal file
@@ -0,0 +1,23 @@
|
||||
name: Close Inactive Issues
|
||||
on:
|
||||
schedule:
|
||||
- cron: "30 1 * * *"
|
||||
|
||||
jobs:
|
||||
close-issues:
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
issues: write
|
||||
pull-requests: write
|
||||
steps:
|
||||
- uses: actions/stale@v5
|
||||
with:
|
||||
only-labels: autoclose
|
||||
days-before-issue-stale: 7
|
||||
days-before-issue-close: 7
|
||||
stale-issue-label: "stale"
|
||||
stale-issue-message: "This issue is stale because it has been open for 7 days with no activity."
|
||||
close-issue-message: "This issue was closed because it has been inactive for 7 days since being marked as stale."
|
||||
days-before-pr-stale: -1
|
||||
days-before-pr-close: -1
|
||||
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
||||
169
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build-pull-request.yml
vendored
Normal file
169
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build-pull-request.yml
vendored
Normal file
@@ -0,0 +1,169 @@
|
||||
on:
|
||||
workflow_call:
|
||||
env:
|
||||
OMP_NUM_THREADS: '10'
|
||||
MKL_THREADING_LAYER: GNU
|
||||
jobs:
|
||||
format:
|
||||
name: Format
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Install clang-format
|
||||
run: |
|
||||
sudo apt-get update -y
|
||||
sudo apt-get install -y wget
|
||||
sudo apt install -y lsb-release wget software-properties-common gnupg
|
||||
wget https://apt.llvm.org/llvm.sh
|
||||
chmod u+x llvm.sh
|
||||
sudo ./llvm.sh 18
|
||||
sudo apt-get install -y git-core clang-format-18
|
||||
- name: Verify clang-format
|
||||
run: |
|
||||
git ls-files | grep -E '\.(cpp|h|cu|cuh)$' | xargs clang-format-18 -i
|
||||
if git diff --quiet; then
|
||||
echo "Formatting OK!"
|
||||
else
|
||||
echo "Formatting not OK!"
|
||||
echo "------------------"
|
||||
git --no-pager diff --color
|
||||
exit 1
|
||||
fi
|
||||
linux-x86_64-cmake:
|
||||
name: Linux x86_64 (cmake)
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
linux-x86_64-AVX2-cmake:
|
||||
name: Linux x86_64 AVX2 (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
opt_level: avx2
|
||||
linux-x86_64-AVX512-cmake:
|
||||
name: Linux x86_64 AVX512 (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: faiss-aws-m7i.large
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
opt_level: avx512
|
||||
linux-x86_64-AVX512_SPR-cmake:
|
||||
name: Linux x86_64 AVX512_SPR (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: faiss-aws-m7i.large
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
opt_level: avx512_spr
|
||||
linux-x86_64-GPU-cmake:
|
||||
name: Linux x86_64 GPU (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
gpu: ON
|
||||
linux-x86_64-GPU-w-CUVS-cmake:
|
||||
name: Linux x86_64 GPU w/ cuVS (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
gpu: ON
|
||||
cuvs: ON
|
||||
linux-x86_64-GPU-w-ROCm-cmake:
|
||||
name: Linux x86_64 GPU w/ ROCm (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: faiss-amd-MI200
|
||||
container:
|
||||
image: ubuntu:22.04
|
||||
options: --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN
|
||||
steps:
|
||||
- name: Container setup
|
||||
run: |
|
||||
if [ -f /.dockerenv ]; then
|
||||
apt-get update && apt-get install -y sudo && apt-get install -y git
|
||||
git config --global --add safe.directory '*'
|
||||
else
|
||||
echo 'Skipping. Current job is not running inside a container.'
|
||||
fi
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
gpu: ON
|
||||
rocm: ON
|
||||
linux-arm64-SVE-cmake:
|
||||
name: Linux arm64 SVE (cmake)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: faiss-aws-r8g.large
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
- name: Build and Test (cmake)
|
||||
uses: ./.github/actions/build_cmake
|
||||
with:
|
||||
opt_level: sve
|
||||
env:
|
||||
# Context: https://github.com/facebookresearch/faiss/wiki/Troubleshooting#surprising-faiss-openmp-and-openblas-interaction
|
||||
OPENBLAS_NUM_THREADS: '1'
|
||||
linux-x86_64-conda:
|
||||
name: Linux x86_64 (conda)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
windows-x86_64-conda:
|
||||
name: Windows x86_64 (conda)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: windows-2019
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
linux-arm64-conda:
|
||||
name: Linux arm64 (conda)
|
||||
needs: linux-x86_64-cmake
|
||||
runs-on: 2-core-ubuntu-arm
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
144
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build-release.yml
vendored
Normal file
144
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build-release.yml
vendored
Normal file
@@ -0,0 +1,144 @@
|
||||
on:
|
||||
workflow_call:
|
||||
secrets:
|
||||
ANACONDA_API_TOKEN:
|
||||
required: true
|
||||
env:
|
||||
OMP_NUM_THREADS: '10'
|
||||
MKL_THREADING_LAYER: GNU
|
||||
jobs:
|
||||
linux-x86_64-packages:
|
||||
name: Linux x86_64 packages
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
linux-x86_64-GPU-packages-CUDA-11-4-4:
|
||||
name: Linux x86_64 GPU packages (CUDA 11.4.4)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "60-real;61-real;62-real;70-real;72-real;75-real;80;86-real"
|
||||
FAISS_FLATTEN_CONDA_INCLUDES: "1"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
cuda: "11.4.4"
|
||||
linux-x86_64-GPU-CUVS-packages-CUDA11-8-0:
|
||||
name: Linux x86_64 GPU w/ cuVS packages (CUDA 11.8.0)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
cuvs: "ON"
|
||||
cuda: "11.8.0"
|
||||
linux-x86_64-GPU-packages-CUDA-12-1-1:
|
||||
name: Linux x86_64 GPU packages (CUDA 12.1.1)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
cuda: "12.1.1"
|
||||
linux-x86_64-GPU-CUVS-packages-CUDA12-4-0:
|
||||
name: Linux x86_64 GPU w/ cuVS packages (CUDA 12.4.0)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
cuvs: "ON"
|
||||
cuda: "12.4.0"
|
||||
windows-x86_64-packages:
|
||||
name: Windows x86_64 packages
|
||||
runs-on: windows-2019
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
osx-arm64-packages:
|
||||
name: OSX arm64 packages
|
||||
runs-on: macos-14
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
linux-arm64-packages:
|
||||
name: Linux arm64 packages
|
||||
runs-on: 2-core-ubuntu-arm
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- name: Build and Package (conda)
|
||||
uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: main
|
||||
17
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build.yml
vendored
Normal file
17
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/build.yml
vendored
Normal file
@@ -0,0 +1,17 @@
|
||||
name: Build
|
||||
on:
|
||||
workflow_dispatch:
|
||||
pull_request:
|
||||
branches:
|
||||
- main
|
||||
push:
|
||||
tags:
|
||||
- 'v*'
|
||||
jobs:
|
||||
build-pull-request:
|
||||
uses: ./.github/workflows/build-pull-request.yml
|
||||
build-release:
|
||||
uses: ./.github/workflows/build-release.yml
|
||||
secrets:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
|
||||
148
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/nightly.yml
vendored
Normal file
148
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/nightly.yml
vendored
Normal file
@@ -0,0 +1,148 @@
|
||||
name: Nightly
|
||||
on:
|
||||
schedule:
|
||||
- cron: '10 6 * * *'
|
||||
env:
|
||||
OMP_NUM_THREADS: '10'
|
||||
MKL_THREADING_LAYER: GNU
|
||||
jobs:
|
||||
linux-x86_64-nightly:
|
||||
name: Linux x86_64 nightlies
|
||||
runs-on: 4-core-ubuntu
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
linux-x86_64-GPU-CUDA-11-4-4-nightly:
|
||||
name: Linux x86_64 GPU nightlies (CUDA 11.4.4)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "60-real;61-real;62-real;70-real;72-real;75-real;80;86-real"
|
||||
FAISS_FLATTEN_CONDA_INCLUDES: "1"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
cuda: "11.4.4"
|
||||
linux-x86_64-GPU-CUVS-CUDA11-8-0-nightly:
|
||||
name: Linux x86_64 GPU w/ cuVS nightlies (CUDA 11.8.0)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
cuvs: "ON"
|
||||
cuda: "11.8.0"
|
||||
linux-x86_64-GPU-CUDA-12-1-1-nightly:
|
||||
name: Linux x86_64 GPU nightlies (CUDA 12.1.1)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
cuda: "12.1.1"
|
||||
linux-x86_64-GPU-CUVS-CUDA12-4-0-nightly:
|
||||
name: Linux x86_64 GPU w/ cuVS nightlies (CUDA 12.4.0)
|
||||
runs-on: 4-core-ubuntu-gpu-t4
|
||||
env:
|
||||
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
cuvs: "ON"
|
||||
cuda: "12.4.0"
|
||||
windows-x86_64-nightly:
|
||||
name: Windows x86_64 nightlies
|
||||
runs-on: windows-2019
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
osx-arm64-nightly:
|
||||
name: OSX arm64 nightlies
|
||||
runs-on: macos-14
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
linux-arm64-nightly:
|
||||
name: Linux arm64 nightlies
|
||||
runs-on: 2-core-ubuntu-arm
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
fetch-tags: true
|
||||
- uses: ./.github/actions/build_conda
|
||||
env:
|
||||
ANACONDA_API_TOKEN: ${{ secrets.ANACONDA_API_TOKEN }}
|
||||
with:
|
||||
label: nightly
|
||||
auto-retry:
|
||||
name: Auto retry on failure
|
||||
if: fromJSON(github.run_attempt) < 2
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Start rerun workflow
|
||||
env:
|
||||
GH_REPO: ${{ github.repository }}
|
||||
GH_TOKEN: ${{ github.token }}
|
||||
GH_DEBUG: api
|
||||
run: |
|
||||
gh workflow run retry_build.yml \
|
||||
-F run_id=${{ github.run_id }}
|
||||
44
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/publish-docs.yml
vendored
Normal file
44
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/publish-docs.yml
vendored
Normal file
@@ -0,0 +1,44 @@
|
||||
name: Publish Docs
|
||||
on:
|
||||
page_build:
|
||||
branches:
|
||||
- gh-pages
|
||||
paths-ignore:
|
||||
- 'docs/**'
|
||||
workflow_run:
|
||||
workflows: [update-doxygen]
|
||||
types:
|
||||
- completed
|
||||
jobs:
|
||||
build_and_publish:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v2
|
||||
with:
|
||||
python-version: 3.8
|
||||
- name: Checkout gh-pages
|
||||
run: |
|
||||
git fetch origin gh-pages
|
||||
git checkout gh-pages
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -r requirements.txt
|
||||
- name: Generate html
|
||||
run: |
|
||||
make html
|
||||
git rm -rf docs
|
||||
mv _build/html docs
|
||||
touch docs/.nojekyll
|
||||
- name: Push changes
|
||||
run: |
|
||||
git config --global user.email "$GITHUB_ACTOR@users.noreply.github.com"
|
||||
git config --global user.name "$GITHUB_ACTOR"
|
||||
git add docs
|
||||
if [ -n "$(git status --porcelain)" ]
|
||||
then
|
||||
git commit docs -m "Sphinx rebuild ($(git rev-parse --short gh-pages))."
|
||||
git push origin gh-pages
|
||||
fi
|
||||
33
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/retry_build.yml
vendored
Normal file
33
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/retry_build.yml
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
name: Retry Build
|
||||
on:
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
run_id:
|
||||
required: true
|
||||
jobs:
|
||||
rerun-on-failure:
|
||||
permissions: write-all
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: rerun ${{ inputs.run_id }}
|
||||
env:
|
||||
GH_REPO: ${{ github.repository }}
|
||||
GH_TOKEN: ${{ github.token }}
|
||||
GH_DEBUG: api
|
||||
run: |
|
||||
# status can be one of "queued", "in_progress", "completed", "waiting", "requested", "pending"
|
||||
# https://docs.github.com/en/rest/checks/runs
|
||||
# while not completed, sleep for 10 minutes
|
||||
while gh run view ${{ inputs.run_id }} --json status | grep -v completed
|
||||
do
|
||||
echo Workflow in progress - sleeping for 10 minutes then checking again
|
||||
sleep 10m
|
||||
done
|
||||
|
||||
# Only retry if there are failed jobs
|
||||
if gh run view ${{ inputs.run_id }} --exit-status; then
|
||||
echo Workflow succeeded - no retry necessary.
|
||||
else
|
||||
echo Workflow failed - initiating retry.
|
||||
gh run rerun ${{ inputs.run_id }} --failed
|
||||
fi
|
||||
40
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/update-doxygen.yml
vendored
Normal file
40
packages/leann-backend-hnsw/third_party/faiss/.github/workflows/update-doxygen.yml
vendored
Normal file
@@ -0,0 +1,40 @@
|
||||
name: Update Doxygen
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
paths:
|
||||
- 'faiss/**'
|
||||
jobs:
|
||||
doxygen:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v2
|
||||
with:
|
||||
python-version: 3.8
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
sudo apt-get install -y doxygen
|
||||
python -m pip install --upgrade pip
|
||||
pip install breathe
|
||||
- name: Generate doxygen xml
|
||||
run: doxygen
|
||||
- name: Push changes
|
||||
run: |
|
||||
git config --global user.email "$GITHUB_ACTOR@users.noreply.github.com"
|
||||
git config --global user.name "$GITHUB_ACTOR"
|
||||
mkdir ./tmp
|
||||
mv xml ./tmp/xml
|
||||
git fetch origin gh-pages
|
||||
git checkout gh-pages
|
||||
git rm -rf xml cpp_api
|
||||
mv ./tmp/xml ./xml
|
||||
breathe-apidoc -o cpp_api xml
|
||||
git add xml cpp_api
|
||||
if [ -n "$(git status --porcelain)" ]
|
||||
then
|
||||
git commit -m "Update API docs ($(git rev-parse --short main))."
|
||||
git push origin gh-pages
|
||||
fi
|
||||
Reference in New Issue
Block a user