Skip to content

Commit

Permalink
pin nightlies with testing on pr
Browse files Browse the repository at this point in the history
  • Loading branch information
junjieqi committed Dec 14, 2024
1 parent 87a760e commit 1fe84ae
Show file tree
Hide file tree
Showing 4 changed files with 28 additions and 26 deletions.
6 changes: 4 additions & 2 deletions .github/workflows/nightly.yml
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
name: Nightly
on:
schedule:
- cron: '10 6 * * *'
workflow_dispatch:
pull_request:
branches:
- main
env:
OMP_NUM_THREADS: '10'
MKL_THREADING_LAYER: GNU
Expand Down
16 changes: 8 additions & 8 deletions conda/faiss-gpu-cuvs/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -46,24 +46,24 @@ outputs:
requirements:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 # [linux64]
- sysroot_linux-64 =2.17 # [linux64]
- llvm-openmp # [osx]
- cmake >=3.26.4
- make # [not win]
- make =4.2 # [not win]
- _openmp_mutex =4.5=2_kmp_llvm # [x86_64]
- mkl =2023 # [x86_64]
- mkl-devel =2023 # [x86_64]
- cuda-toolkit {{ cudatoolkit }}
host:
- _openmp_mutex =4.5=2_kmp_llvm # [x86_64]
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
- libcuvs =24.08
- cuda-version {{ cudatoolkit }}
run:
- _openmp_mutex =4.5=2_kmp_llvm # [x86_64]
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
- cuda-cudart {{ cuda_constraints }}
- libcublas {{ libcublas_constraints }}
- libcuvs =24.08
Expand All @@ -88,9 +88,9 @@ outputs:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 =2.17 # [linux64]
- swig
- swig =4.0
- cmake >=3.24.0
- make # [not win]
- make =4.2 # [not win]
- _openmp_mutex =4.5=2_kmp_llvm # [x86_64]
- mkl =2023 # [x86_64]
- cuda-toolkit {{ cudatoolkit }}
Expand All @@ -107,9 +107,9 @@ outputs:
- {{ pin_subpackage('libfaiss', exact=True) }}
test:
requires:
- numpy
- numpy >=1.19,<2
- scipy
- pytorch
- pytorch <2.5
- pytorch-cuda {{ cuda_constraints }}
commands:
- python -X faulthandler -m unittest discover -v -s tests/ -p "test_*"
Expand Down
16 changes: 8 additions & 8 deletions conda/faiss-gpu/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -47,18 +47,18 @@ outputs:
requirements:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 # [linux64]
- sysroot_linux-64 =2.17 # [linux64]
- llvm-openmp # [osx]
- cmake >=3.24.0
- make # [not win]
- make =4.2 # [not win]
- mkl-devel =2023 # [x86_64]
- cuda-toolkit {{ cudatoolkit }}
host:
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
run:
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
- cuda-cudart {{ cuda_constraints }}
- libcublas {{ libcublas_constraints }}
test:
Expand All @@ -81,9 +81,9 @@ outputs:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 =2.17 # [linux64]
- swig
- swig =4.0
- cmake >=3.24.0
- make # [not win]
- make =4.2 # [not win]
- cuda-toolkit {{ cudatoolkit }}
host:
- python {{ python }}
Expand All @@ -96,9 +96,9 @@ outputs:
- {{ pin_subpackage('libfaiss', exact=True) }}
test:
requires:
- numpy
- numpy >=1.19,<2
- scipy
- pytorch
- pytorch <2.5
- pytorch-cuda {{ cuda_constraints }}
commands:
- python -X faulthandler -m unittest discover -v -s tests/ -p "test_*"
Expand Down
16 changes: 8 additions & 8 deletions conda/faiss/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -37,17 +37,17 @@ outputs:
requirements:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 # [linux64]
- sysroot_linux-64 =2.17 # [linux64]
- llvm-openmp # [osx]
- cmake >=3.24.0
- make # [not win]
- make =4.2 # [not win]
- mkl-devel =2023 # [x86_64]
host:
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
run:
- mkl =2023 # [x86_64]
- openblas # [not x86_64]
- openblas =0.3 # [not x86_64]
test:
requires:
- conda-build
Expand All @@ -68,9 +68,9 @@ outputs:
build:
- {{ compiler('cxx') }}
- sysroot_linux-64 =2.17 # [linux64]
- swig
- swig =4.0
- cmake >=3.24.0
- make # [not win]
- make =4.2 # [not win]
host:
- python {{ python }}
- numpy >=1.19,<2
Expand All @@ -82,9 +82,9 @@ outputs:
- {{ pin_subpackage('libfaiss', exact=True) }}
test:
requires:
- numpy
- numpy >=1.19,<2
- scipy
- pytorch
- pytorch <2.5
commands:
- python -X faulthandler -m unittest discover -v -s tests/ -p "test_*"
- python -X faulthandler -m unittest discover -v -s tests/ -p "torch_*"
Expand Down

0 comments on commit 1fe84ae

Please sign in to comment.