Benchmark TPU Nightly #12
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Benchmark TPU Nightly | |
| on: | |
| push: # TODO: remove before merging — temporary trigger for CI testing | |
| branches: | |
| - yifeixu/tpu-nightly-benchmark | |
| schedule: | |
| - cron: '0 10 * * *' # Runs at 2 AM PST (10 AM UTC) | |
| workflow_dispatch: | |
| inputs: | |
| kernels: | |
| description: 'Comma-separated list of kernels to benchmark' | |
| required: false | |
| type: string | |
| # Excluded kernels: | |
| # layer_norm: OOB slice when reduction_loops doesn't evenly divide the reduction dim (gh#1937) | |
| default: "exp,add,softmax_two_pass,welford,attention,bmm,geglu,grpo_loss,jagged_hstu_attn,low_mem_dropout,swiglu" | |
| permissions: | |
| contents: read | |
| jobs: | |
| benchmark-tpu: | |
| uses: ./.github/workflows/benchmark_tpu.yml | |
| permissions: | |
| contents: read | |
| with: | |
| kernels: ${{ github.event.inputs.kernels || 'exp,add,softmax_two_pass,welford,attention,bmm,geglu,grpo_loss,jagged_hstu_attn,low_mem_dropout,swiglu' }} |