mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
workflow to run sdxl with search (#10543)
This commit is contained in:
21
.github/workflows/benchmark_search.yml
vendored
Normal file
21
.github/workflows/benchmark_search.yml
vendored
Normal 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
|
||||
Reference in New Issue
Block a user