1
1
name : Fast GPU Tests on main
2
2
3
3
on :
4
+ pull_request :
5
+ branches : main
6
+ paths :
7
+ - " src/diffusers/models/modeling_utils.py"
8
+ - " src/diffusers/models/model_loading_utils.py"
9
+ - " src/diffusers/pipelines/pipeline_utils.py"
10
+ - " src/diffusers/pipeline_loading_utils.py"
4
11
workflow_dispatch :
5
12
push :
6
13
branches :
@@ -160,6 +167,7 @@ jobs:
160
167
path : reports
161
168
162
169
flax_tpu_tests :
170
+ if : ${{ github.event_name != 'pull_request' }}
163
171
name : Flax TPU Tests
164
172
runs-on :
165
173
group : gcp-ct5lp-hightpu-8t
@@ -208,6 +216,7 @@ jobs:
208
216
path : reports
209
217
210
218
onnx_cuda_tests :
219
+ if : ${{ github.event_name != 'pull_request' }}
211
220
name : ONNX CUDA Tests
212
221
runs-on :
213
222
group : aws-g4dn-2xlarge
@@ -256,6 +265,7 @@ jobs:
256
265
path : reports
257
266
258
267
run_torch_compile_tests :
268
+ if : ${{ github.event_name != 'pull_request' }}
259
269
name : PyTorch Compile CUDA tests
260
270
261
271
runs-on :
@@ -299,6 +309,7 @@ jobs:
299
309
path : reports
300
310
301
311
run_xformers_tests :
312
+ if : ${{ github.event_name != 'pull_request' }}
302
313
name : PyTorch xformers CUDA tests
303
314
304
315
runs-on :
@@ -349,7 +360,6 @@ jobs:
349
360
container :
350
361
image : diffusers/diffusers-pytorch-cuda
351
362
options : --gpus 0 --shm-size "16gb" --ipc host
352
-
353
363
steps :
354
364
- name : Checkout diffusers
355
365
uses : actions/checkout@v3
@@ -359,7 +369,6 @@ jobs:
359
369
- name : NVIDIA-SMI
360
370
run : |
361
371
nvidia-smi
362
-
363
372
- name : Install dependencies
364
373
run : |
365
374
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
0 commit comments