-
Notifications
You must be signed in to change notification settings - Fork 14
42 lines (40 loc) · 1.22 KB
/
Copy pathci.yml
File metadata and controls
42 lines (40 loc) · 1.22 KB
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
name: CI
on: push
jobs:
build:
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v2
with:
submodules: 'recursive'
- uses: Jimver/cuda-toolkit@v0.2.8
id: cuda-toolkit
with:
cuda: '11.3.1'
method: network
- name: Install other dependencies
run: |
sudo apt-get update
sudo apt-get install -y build-essential libcublas-dev-11-3 libcudnn8-dev tensorrt-dev
- uses: conda-incubator/setup-miniconda@v2
with:
activate-environment: nart
environment-file: .github/conda/build-environment.yml
auto-activate-base: false
use-only-tar-bz2: true
python-version: 3.7
- name: Conda info
shell: bash -l {0}
run: conda info
- name: Build NART
shell: bash -l {0}
run: |
cmake -B build -DNART_CASE_MODULES='quant;cuda;tensorrt;cuda_quant' -DENABLE_NART_TOOLS=ON
cmake --build build
- name: Install NART
shell: bash -l {0}
run: |
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib
cmake --install build --prefix $CONDA_PREFIX
cd python
python setup.py install