-
Notifications
You must be signed in to change notification settings - Fork 239
173 lines (166 loc) · 5.32 KB
/
precommit.yml
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
name: precommit
on:
pull_request:
types:
- opened
- reopened
- synchronize
paths-ignore:
- '**/*.md'
- 'docs/**/*'
jobs:
# common:
# runs-on: ubuntu-20.04
# steps:
# - uses: actions/checkout@v3
# with:
# lfs: true
# - uses: actions/setup-python@v3
# with:
# python-version: 3.8.10
# - name: Install NNCF and test requirements
# run: make install-common-test
# - name: Run common precommit test scope
# run: make test-common
# env:
# NNCF_COVERAGE: 1
# - name: Upload coverage report as artifact
# uses: actions/upload-artifact@v3
# with:
# name: coverage_common
# path: ./coverage.xml
# - name: Upload coverage report to codecov
# uses: codecov/codecov-action@v3
# with:
# token: ${{ secrets.CODECOV_TOKEN }}
# name: coverage_common
# flags: COMMON
# onnx:
# runs-on: ubuntu-20.04
# steps:
# - uses: actions/checkout@v3
# with:
# lfs: true
# - uses: actions/setup-python@v3
# with:
# python-version: 3.8.10
# - name: Install NNCF and test requirements
# run: make install-onnx-test
# - name: Run ONNX precommit test scope
# run: make test-onnx
# env:
# NNCF_COVERAGE: 1
# - name: Upload coverage report as artifact
# uses: actions/upload-artifact@v3
# with:
# name: coverage_onnx
# path: ./coverage.xml
# - name: Upload coverage report to codecov
# uses: codecov/codecov-action@v3
# with:
# token: ${{ secrets.CODECOV_TOKEN }}
# name: coverage_onnx
# flags: ONNX
# openvino:
# runs-on: ubuntu-20.04
# steps:
# - uses: actions/checkout@v3
# with:
# lfs: true
# - uses: actions/setup-python@v3
# with:
# python-version: 3.8.10
# - name: Install NNCF and test requirements
# run: make install-openvino-test
# - name: Run OV precommit test scope
# run: make test-openvino
# env:
# NNCF_COVERAGE: 1
# - name: Upload coverage report as artifact
# uses: actions/upload-artifact@v3
# with:
# name: coverage_openvino
# path: ./coverage.xml
# - name: Upload coverage report to codecov
# uses: codecov/codecov-action@v3
# with:
# token: ${{ secrets.CODECOV_TOKEN }}
# name: coverage_openvino
# flags: OPENVINO
pytorch-cpu:
timeout-minutes: 40
defaults:
run:
shell: bash
runs-on: ubuntu-20.04-8-cores # aks-linux-16-cores-32gb
steps:
- name: Install dependencies
run : |
sudo apt-get update
sudo apt-get --assume-yes install gcc g++ build-essential ninja-build libgl1-mesa-dev libglib2.0-0
- uses: actions/checkout@v3
with:
lfs: true
- uses: actions/setup-python@v3
with:
python-version: 3.8.10
- name: Runner info
continue-on-error: true
run: |
cat /etc/*release
cat /proc/cpuinfo
- name: Install NNCF and test requirements
run: make install-torch-test
- name: Run PyTorch precommit test scope
run: |
pytest tests/torch -n4 -ra -m "not cuda and not weekly and not nightly and not models_hub"
# - name: Runner info
# continue-on-error: true
# run: |
# cat /proc/cpuinfo
# cat /etc/*release
# - name: TEST
# run: |
# pip install -e .
# pip install torch==2.2.1
# python --version
# python -c "from nncf.torch.quantization.extensions import QuantizedFunctionsCPULoader; QuantizedFunctionsCPULoader.load()"
pytorch-cuda:
timeout-minutes: 40
defaults:
run:
shell: bash
runs-on: aks-linux-4-cores-28gb-gpu-testla-t4
steps:
- name: Install dependencies
run : |
sudo apt-get update
sudo apt-get --assume-yes install gcc ninja-build libgl1-mesa-dev libglib2.0-0
- name: Download CUDA
run: |
wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run --toolkit --silent
- name: Runner info
continue-on-error: true
run: |
export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
nvidia-smi
cat /proc/cpuinfo
nvcc --version
- uses: actions/checkout@v3
with:
lfs: true
- uses: actions/setup-python@v3
with:
python-version: 3.8.10
- name: Install NNCF and test requirements
run: make install-torch-test
- name: Check CUDA
run: |
python -c "import torch; print(torch.cuda.is_available())"
- name: Run PyTorch precommit test scope
run: |
export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
pytest tests/torch -ra -m "cuda and not weekly and not nightly and not models_hub"