workflow to run sdxl with search (#10543)

This commit is contained in:
chenyu
2025-05-27 17:25:41 -04:00
committed by GitHub
parent d1f0043331
commit fffdc4d31c

21
.github/workflows/benchmark_search.yml vendored Normal file
View File

@@ -0,0 +1,21 @@
name: Benchmark with kernel search
on:
push:
branches:
- update_benchmark_search
workflow_dispatch:
jobs:
run_script_job:
runs-on: [self-hosted, Linux, tinybox]
if: github.repository_owner == 'tinygrad'
steps:
- name: Checkout Code
uses: actions/checkout@v4
- name: Remove amdgpu
run: sudo rmmod amdgpu || true
- name: Run SDXL
run: |
BENCHMARK_LOG=search_sdxl PYTHONPATH=. AMD=1 JITBEAM=2 IGNORE_BEAM_CACHE=1 python examples/sdxl.py --noshow --timing --seed 0