Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 11 additions & 2 deletions .github/workflows/push_tests.yml
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
name: Fast GPU Tests on main

on:
pull_request:
branches: main
paths:
- "src/diffusers/models/modeling_utils.py"
- "src/diffusers/models/model_loading_utils.py"
- "src/diffusers/pipelines/pipeline_utils.py"
- "src/diffusers/pipeline_loading_utils.py"
workflow_dispatch:
push:
branches:
Expand Down Expand Up @@ -160,6 +167,7 @@ jobs:
path: reports

flax_tpu_tests:
if: ${{ github.event_name != 'pull_request' }}
name: Flax TPU Tests
runs-on:
group: gcp-ct5lp-hightpu-8t
Expand Down Expand Up @@ -208,6 +216,7 @@ jobs:
path: reports

onnx_cuda_tests:
if: ${{ github.event_name != 'pull_request' }}
name: ONNX CUDA Tests
runs-on:
group: aws-g4dn-2xlarge
Expand Down Expand Up @@ -256,6 +265,7 @@ jobs:
path: reports

run_torch_compile_tests:
if: ${{ github.event_name != 'pull_request' }}
name: PyTorch Compile CUDA tests

runs-on:
Expand Down Expand Up @@ -299,6 +309,7 @@ jobs:
path: reports

run_xformers_tests:
if: ${{ github.event_name != 'pull_request' }}
name: PyTorch xformers CUDA tests

runs-on:
Expand Down Expand Up @@ -349,7 +360,6 @@ jobs:
container:
image: diffusers/diffusers-pytorch-cuda
options: --gpus 0 --shm-size "16gb" --ipc host

steps:
- name: Checkout diffusers
uses: actions/checkout@v3
Expand All @@ -359,7 +369,6 @@ jobs:
- name: NVIDIA-SMI
run: |
nvidia-smi

- name: Install dependencies
run: |
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
Expand Down
Loading