Skip to content

Commit ca4b34f

Browse files
authored
Add int4_gemm (#917)
1 parent ed3c099 commit ca4b34f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/benchmark_dispatch.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ on:
2222
description: 'Comma-separated list of kernels to benchmark'
2323
required: false
2424
type: string
25-
default: "softmax,jsd,welford,kl_div,layer_norm,layer_norm-bwd,rms_norm,rms_norm-bwd,cross_entropy,flash_attention,gemm,grouped_gemm"
25+
default: "softmax,jsd,welford,kl_div,int4_gemm,layer_norm,layer_norm-bwd,rms_norm,rms_norm-bwd,cross_entropy,flash_attention,gemm,grouped_gemm"
2626

2727
jobs:
2828
gen-matrix-h100:
2929
if: ${{ github.event.inputs.run_h100 == 'true' }}
3030
uses: ./.github/workflows/compute-benchmark-matrix.yml
3131
with:
32-
max-runners: 12
32+
max-runners: 13
3333
kernels: ${{ github.event.inputs.kernels }}
3434

3535
run-h100:
@@ -54,7 +54,7 @@ jobs:
5454
uses: ./.github/workflows/compute-benchmark-matrix.yml
5555
if: ${{ github.event.inputs.run_b200 == 'true' }}
5656
with:
57-
max-runners: 12
57+
max-runners: 13
5858
kernels: ${{ github.event.inputs.kernels }}
5959

6060
run-b200:

0 commit comments

Comments
 (0)