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