Unverified Commit 454f82e6 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[CI] run fast gpu tests conditionally on pull requests. (#10310)

* run fast gpu tests conditionally on pull requests.

* revert unneeded changes.

* simplify PR.
parent 1f853504
name: Fast GPU Tests on main name: Fast GPU Tests on main
on: 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: workflow_dispatch:
push: push:
branches: branches:
...@@ -160,6 +167,7 @@ jobs: ...@@ -160,6 +167,7 @@ jobs:
path: reports path: reports
flax_tpu_tests: flax_tpu_tests:
if: ${{ github.event_name != 'pull_request' }}
name: Flax TPU Tests name: Flax TPU Tests
runs-on: runs-on:
group: gcp-ct5lp-hightpu-8t group: gcp-ct5lp-hightpu-8t
...@@ -208,6 +216,7 @@ jobs: ...@@ -208,6 +216,7 @@ jobs:
path: reports path: reports
onnx_cuda_tests: onnx_cuda_tests:
if: ${{ github.event_name != 'pull_request' }}
name: ONNX CUDA Tests name: ONNX CUDA Tests
runs-on: runs-on:
group: aws-g4dn-2xlarge group: aws-g4dn-2xlarge
...@@ -256,6 +265,7 @@ jobs: ...@@ -256,6 +265,7 @@ jobs:
path: reports path: reports
run_torch_compile_tests: run_torch_compile_tests:
if: ${{ github.event_name != 'pull_request' }}
name: PyTorch Compile CUDA tests name: PyTorch Compile CUDA tests
runs-on: runs-on:
...@@ -299,6 +309,7 @@ jobs: ...@@ -299,6 +309,7 @@ jobs:
path: reports path: reports
run_xformers_tests: run_xformers_tests:
if: ${{ github.event_name != 'pull_request' }}
name: PyTorch xformers CUDA tests name: PyTorch xformers CUDA tests
runs-on: runs-on:
...@@ -349,7 +360,6 @@ jobs: ...@@ -349,7 +360,6 @@ jobs:
container: container:
image: diffusers/diffusers-pytorch-cuda image: diffusers/diffusers-pytorch-cuda
options: --gpus 0 --shm-size "16gb" --ipc host options: --gpus 0 --shm-size "16gb" --ipc host
steps: steps:
- name: Checkout diffusers - name: Checkout diffusers
uses: actions/checkout@v3 uses: actions/checkout@v3
...@@ -359,7 +369,6 @@ jobs: ...@@ -359,7 +369,6 @@ jobs:
- name: NVIDIA-SMI - name: NVIDIA-SMI
run: | run: |
nvidia-smi nvidia-smi
- name: Install dependencies - name: Install dependencies
run: | run: |
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH" python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment