Skip to content

Commit 6f75313

Browse files
mengdilinmengdilin
committed
openmp with libmam-solver
1 parent e8edc6b commit 6f75313

File tree

2 files changed

+165
-4
lines changed

2 files changed

+165
-4
lines changed

‎.github/actions/build_cmake/action.yml

+1-3
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ runs:
3636
3737
# install base packages for ARM64
3838
if [ "${{ runner.arch }}" = "ARM64" ]; then
39+
conda install --solver=classic conda-forge::conda-libmamba-solver conda-forge::libmamba conda-forge::libmambapy conda-forge::libarchive
3940
conda install -y -q -c conda-forge openblas=*=*openmp* gxx_linux-aarch64 sysroot_linux-aarch64
4041
fi
4142
@@ -152,9 +153,6 @@ runs:
152153
run: |
153154
pytest --junitxml=test-results/pytest/results.xml tests/test_*.py
154155
pytest --junitxml=test-results/pytest/results-torch.xml tests/torch_*.py
155-
- name: Setup tmate session
156-
if: failure()
157-
uses: mxschmitt/action-tmate@v3
158156
- name: Python tests (CPU + GPU)
159157
if: inputs.gpu == 'ON'
160158
shell: bash

‎.github/workflows/build.yml

+164-1
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,167 @@ jobs:
2020
- name: Build and Test (cmake)
2121
uses: ./.github/actions/build_cmake
2222
with:
23-
opt_level: sve
23+
opt_level: sve
24+
linux-x86_64-conda:
25+
name: Linux x86_64 (conda)
26+
needs: linux-x86_64-cmake
27+
runs-on: ubuntu-latest
28+
steps:
29+
- name: Checkout
30+
uses: actions/checkout@v4
31+
with:
32+
fetch-depth: 0
33+
fetch-tags: true
34+
- name: Build and Package (conda)
35+
uses: ./.github/actions/build_conda
36+
windows-x86_64-conda:
37+
name: Windows x86_64 (conda)
38+
needs: linux-x86_64-cmake
39+
runs-on: windows-2019
40+
steps:
41+
- name: Checkout
42+
uses: actions/checkout@v4
43+
with:
44+
fetch-depth: 0
45+
fetch-tags: true
46+
- name: Build and Package (conda)
47+
uses: ./.github/actions/build_conda
48+
linux-arm64-conda:
49+
name: Linux arm64 (conda)
50+
needs: linux-x86_64-cmake
51+
runs-on: 2-core-ubuntu-arm
52+
steps:
53+
- name: Checkout
54+
uses: actions/checkout@v4
55+
with:
56+
fetch-depth: 0
57+
fetch-tags: true
58+
- name: Build and Package (conda)
59+
uses: ./.github/actions/build_conda
60+
linux-x86_64-packages:
61+
name: Linux x86_64 packages
62+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
63+
runs-on: ubuntu-latest
64+
steps:
65+
- name: Checkout
66+
uses: actions/checkout@v4
67+
with:
68+
fetch-depth: 0
69+
fetch-tags: true
70+
- name: Build and Package (conda)
71+
uses: ./.github/actions/build_conda
72+
with:
73+
label: main
74+
linux-x86_64-GPU-packages-CUDA-11-4-4:
75+
name: Linux x86_64 GPU packages (CUDA 11.4.4)
76+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
77+
runs-on: 4-core-ubuntu-gpu-t4
78+
env:
79+
CUDA_ARCHS: "60-real;61-real;62-real;70-real;72-real;75-real;80;86-real"
80+
FAISS_FLATTEN_CONDA_INCLUDES: "1"
81+
steps:
82+
- name: Checkout
83+
uses: actions/checkout@v4
84+
with:
85+
fetch-depth: 0
86+
fetch-tags: true
87+
- name: Build and Package (conda)
88+
uses: ./.github/actions/build_conda
89+
with:
90+
label: main
91+
cuda: "11.4.4"
92+
linux-x86_64-GPU-RAFT-packages-CUDA11-8-0:
93+
name: Linux x86_64 GPU w/ RAFT packages (CUDA 11.8.0)
94+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
95+
runs-on: 4-core-ubuntu-gpu-t4
96+
env:
97+
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
98+
steps:
99+
- name: Checkout
100+
uses: actions/checkout@v4
101+
with:
102+
fetch-depth: 0
103+
fetch-tags: true
104+
- name: Build and Package (conda)
105+
uses: ./.github/actions/build_conda
106+
with:
107+
label: main
108+
raft: "ON"
109+
cuda: "11.8.0"
110+
linux-x86_64-GPU-packages-CUDA-12-1-1:
111+
name: Linux x86_64 GPU packages (CUDA 12.1.1)
112+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
113+
runs-on: 4-core-ubuntu-gpu-t4
114+
env:
115+
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
116+
steps:
117+
- name: Checkout
118+
uses: actions/checkout@v4
119+
with:
120+
fetch-depth: 0
121+
fetch-tags: true
122+
- name: Build and Package (conda)
123+
uses: ./.github/actions/build_conda
124+
with:
125+
label: main
126+
cuda: "12.1.1"
127+
linux-x86_64-GPU-RAFT-packages-CUDA12-1-1:
128+
name: Linux x86_64 GPU w/ RAFT packages (CUDA 12.1.1)
129+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
130+
runs-on: 4-core-ubuntu-gpu-t4
131+
env:
132+
CUDA_ARCHS: "70-real;72-real;75-real;80;86-real"
133+
steps:
134+
- name: Checkout
135+
uses: actions/checkout@v4
136+
with:
137+
fetch-depth: 0
138+
fetch-tags: true
139+
- name: Build and Package (conda)
140+
uses: ./.github/actions/build_conda
141+
with:
142+
label: main
143+
raft: "ON"
144+
cuda: "12.1.1"
145+
windows-x86_64-packages:
146+
name: Windows x86_64 packages
147+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
148+
runs-on: windows-2019
149+
steps:
150+
- name: Checkout
151+
uses: actions/checkout@v4
152+
with:
153+
fetch-depth: 0
154+
fetch-tags: true
155+
- name: Build and Package (conda)
156+
uses: ./.github/actions/build_conda
157+
with:
158+
label: main
159+
osx-arm64-packages:
160+
name: OSX arm64 packages
161+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
162+
runs-on: macos-14
163+
steps:
164+
- name: Checkout
165+
uses: actions/checkout@v4
166+
with:
167+
fetch-depth: 0
168+
fetch-tags: true
169+
- name: Build and Package (conda)
170+
uses: ./.github/actions/build_conda
171+
with:
172+
label: main
173+
linux-arm64-packages:
174+
name: Linux arm64 packages
175+
if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
176+
runs-on: 2-core-ubuntu-arm
177+
steps:
178+
- name: Checkout
179+
uses: actions/checkout@v4
180+
with:
181+
fetch-depth: 0
182+
fetch-tags: true
183+
- name: Build and Package (conda)
184+
uses: ./.github/actions/build_conda
185+
with:
186+
label: main

0 commit comments

Comments
 (0)