Skip to content

Commit 52a340a

Browse files
committed
revert unneeded changes.
1 parent e64b90e commit 52a340a

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

.github/workflows/push_tests.yml

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ jobs:
9191
- name: PyTorch CUDA checkpoint tests on Ubuntu
9292
env:
9393
HF_TOKEN: ${{ secrets.HF_TOKEN }}
94+
# https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
9495
CUBLAS_WORKSPACE_CONFIG: :16:8
9596
run: |
9697
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
@@ -144,6 +145,7 @@ jobs:
144145
- name: Run PyTorch CUDA tests
145146
env:
146147
HF_TOKEN: ${{ secrets.HF_TOKEN }}
148+
# https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
147149
CUBLAS_WORKSPACE_CONFIG: :16:8
148150
run: |
149151
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
@@ -265,16 +267,20 @@ jobs:
265267
run_torch_compile_tests:
266268
if: ${{ github.event_name != 'pull_request' }}
267269
name: PyTorch Compile CUDA tests
270+
268271
runs-on:
269272
group: aws-g4dn-2xlarge
273+
270274
container:
271275
image: diffusers/diffusers-pytorch-compile-cuda
272276
options: --gpus 0 --shm-size "16gb" --ipc host
277+
273278
steps:
274279
- name: Checkout diffusers
275280
uses: actions/checkout@v3
276281
with:
277282
fetch-depth: 2
283+
278284
- name: NVIDIA-SMI
279285
run: |
280286
nvidia-smi
@@ -294,6 +300,7 @@ jobs:
294300
- name: Failure short reports
295301
if: ${{ failure() }}
296302
run: cat reports/tests_torch_compile_cuda_failures_short.txt
303+
297304
- name: Test suite reports artifacts
298305
if: ${{ always() }}
299306
uses: actions/upload-artifact@v4
@@ -304,16 +311,20 @@ jobs:
304311
run_xformers_tests:
305312
if: ${{ github.event_name != 'pull_request' }}
306313
name: PyTorch xformers CUDA tests
314+
307315
runs-on:
308316
group: aws-g4dn-2xlarge
317+
309318
container:
310319
image: diffusers/diffusers-pytorch-xformers-cuda
311320
options: --gpus 0 --shm-size "16gb" --ipc host
321+
312322
steps:
313323
- name: Checkout diffusers
314324
uses: actions/checkout@v3
315325
with:
316326
fetch-depth: 2
327+
317328
- name: NVIDIA-SMI
318329
run: |
319330
nvidia-smi
@@ -332,6 +343,7 @@ jobs:
332343
- name: Failure short reports
333344
if: ${{ failure() }}
334345
run: cat reports/tests_torch_xformers_cuda_failures_short.txt
346+
335347
- name: Test suite reports artifacts
336348
if: ${{ always() }}
337349
uses: actions/upload-artifact@v4
@@ -340,10 +352,11 @@ jobs:
340352
path: reports
341353

342354
run_examples_tests:
343-
if: ${{ github.event_name != 'pull_request' }}
344355
name: Examples PyTorch CUDA tests on Ubuntu
356+
345357
runs-on:
346358
group: aws-g4dn-2xlarge
359+
347360
container:
348361
image: diffusers/diffusers-pytorch-cuda
349362
options: --gpus 0 --shm-size "16gb" --ipc host
@@ -352,27 +365,34 @@ jobs:
352365
uses: actions/checkout@v3
353366
with:
354367
fetch-depth: 2
368+
355369
- name: NVIDIA-SMI
356370
run: |
357371
nvidia-smi
358372
- name: Install dependencies
359373
run: |
360374
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
361375
python -m uv pip install -e [quality,test,training]
376+
362377
- name: Environment
363378
run: |
379+
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
364380
python utils/print_env.py
381+
365382
- name: Run example tests on GPU
366383
env:
367384
HF_TOKEN: ${{ secrets.HF_TOKEN }}
368385
run: |
386+
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
369387
python -m uv pip install timm
370388
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v --make-reports=examples_torch_cuda examples/
389+
371390
- name: Failure short reports
372391
if: ${{ failure() }}
373392
run: |
374393
cat reports/examples_torch_cuda_stats.txt
375394
cat reports/examples_torch_cuda_failures_short.txt
395+
376396
- name: Test suite reports artifacts
377397
if: ${{ always() }}
378398
uses: actions/upload-artifact@v4

0 commit comments

Comments
 (0)