galaxy/faiss
1
0
Fork 0
mirror of https://github.com/facebookresearch/faiss.git synced 2025-06-03 03:25:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Parameterize CUDA_ARCHS in packaging jobs. (#1671)

Browse Source 
Summary:
This will allow us to support compute capabilities 8.0 and 8.6 (for
Ampere devices) with CUDA 11.

Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1671

Reviewed By: mdouze

Differential Revision: D26338700

Pulled By: beauby

fbshipit-source-id: f023e7a37504d79ab78a45319e5a9cb825e7604a
This commit is contained in:
Lucas Hosseini 2021-02-09 07:35:57 -08:00 committed by Facebook GitHub Bot
parent 10c8583b2d
commit f5a8c29c57
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.circleci/config.yml
View File

@ -144,6 +144,8 @@ jobs:
default: main
cuda:
type: string
cuda_archs:
type: string
machine:
resource_class: gpu.nvidia.small
image: ubuntu-1604-cuda-10.1:201909-23
@ -155,7 +157,7 @@ jobs:
command: |
docker build -t faiss -f conda/Dockerfile.cuda<<parameters.cuda>> .
docker run --gpus all \
-e CUDA_ARCHS="35;52;60;61;70;72;75" \
-e CUDA_ARCHS="<<parameters.cuda_archs>>" \
-e ANACONDA_API_TOKEN=$ANACONDA_API_TOKEN \
faiss \
conda build faiss-gpu --variants '{ "cudatoolkit": "<<parameters.cuda>>" }' \
@ -234,6 +236,7 @@ workflows:
- deploy_linux_gpu:
name: Linux GPU packages (CUDA 10.1)
cuda: "10.1"
cuda_archs: "35;52;60;61;70;72;75"
filters:
tags:
only: /^v.*/
@ -242,6 +245,7 @@ workflows:
- deploy_linux_gpu:
name: Linux GPU packages (CUDA 10.2)
cuda: "10.2"
cuda_archs: "35;52;60;61;70;72;75"
filters:
tags:
only: /^v.*/
@ -250,6 +254,7 @@ workflows:
- deploy_linux_gpu:
name: Linux GPU packages (CUDA 11.0)
cuda: "11.0"
cuda_archs: "35;52;60;61;70;72;75;80;86"
filters:
tags:
only: /^v.*/
@ -285,14 +290,17 @@ workflows:
- deploy_linux_gpu:
name: Linux GPU nightlies (CUDA 10.1)
cuda: "10.1"
cuda_archs: "35;52;60;61;70;72;75"
label: nightly
- deploy_linux_gpu:
name: Linux GPU nightlies (CUDA 10.2)
cuda: "10.2"
cuda_archs: "35;52;60;61;70;72;75"
label: nightly
- deploy_linux_gpu:
name: Linux GPU nightlies (CUDA 11.0)
cuda: "11.0"
cuda_archs: "35;52;60;61;70;72;75;80;86"
label: nightly
- deploy_windows:
name: Windows nightlies