faiss/.github/workflows/build-pip-gpu.yml at 1b8df4a4592cc955c66cd9d50ad06dd7c8409cd2 · facebookresearch/faiss · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

name: Build GPU pip wheels

on:
  workflow_call:
  workflow_dispatch:
  pull_request:
    branches:
      - main
    paths:
      - 'pyproject-gpu.toml'
      - 'pyproject-gpu-cuvs.toml'
      - 'faiss/python/**'
      - 'faiss/gpu/**'
      - '.github/workflows/build-pip-gpu.yml'

jobs:
  build-faiss-gpu:
    name: Build faiss-gpu wheels
    runs-on: 4-core-ubuntu-gpu-t4
    env:
      CIBW_BUILD_VERBOSITY: 1
    steps:
      - name: Checkout
        uses: actions/checkout@v4
        with:
          fetch-depth: 0
          fetch-tags: true

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.12'

      - name: Install cibuildwheel
        run: pip install cibuildwheel==2.22.0

      - name: Activate GPU pyproject.toml
        run: cp pyproject-gpu.toml pyproject.toml

      - name: Build wheels
        run: cibuildwheel --output-dir wheelhouse

      - name: Upload wheels
        uses: actions/upload-artifact@v4
        with:
          name: wheels-gpu
          path: wheelhouse/*.whl

  build-faiss-gpu-cuvs:
    name: Build faiss-gpu-cuvs wheels
    runs-on: 4-core-ubuntu-gpu-t4
    env:
      CIBW_BUILD_VERBOSITY: 1
    steps:
      - name: Checkout
        uses: actions/checkout@v4
        with:
          fetch-depth: 0
          fetch-tags: true

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.12'

      - name: Install cibuildwheel
        run: pip install cibuildwheel==2.22.0

      - name: Activate cuVS pyproject.toml
        run: cp pyproject-gpu-cuvs.toml pyproject.toml

      - name: Build wheels
        run: cibuildwheel --output-dir wheelhouse

      - name: Upload wheels
        uses: actions/upload-artifact@v4
        with:
          name: wheels-gpu-cuvs
          path: wheelhouse/*.whl

  publish-gpu:
    name: Publish faiss-gpu to PyPI
    needs: [build-faiss-gpu]
    runs-on: ubuntu-latest
    if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
    environment:
      name: pypi
      url: https://pypi.org/p/faiss-gpu
    permissions:
      id-token: write
    steps:
      - name: Download wheels
        uses: actions/download-artifact@v4
        with:
          name: wheels-gpu
          path: dist

      - name: Publish to PyPI
        uses: pypa/gh-action-pypi-publish@release/v1

  publish-gpu-cuvs:
    name: Publish faiss-gpu-cuvs to PyPI
    needs: [build-faiss-gpu-cuvs]
    runs-on: ubuntu-latest
    if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
    environment:
      name: pypi
      url: https://pypi.org/p/faiss-gpu-cuvs
    permissions:
      id-token: write
    steps:
      - name: Download wheels
        uses: actions/download-artifact@v4
        with:
          name: wheels-gpu-cuvs
          path: dist

      - name: Publish to PyPI
        uses: pypa/gh-action-pypi-publish@release/v1