Skip to content

Commit e9d62d0

Browse files
fbarchardxnnpack-bot
authored andcommitted
gemm benchmark documentation fix
- update names of models to match files PiperOrigin-RevId: 917984924
1 parent ce14e18 commit e9d62d0

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

bench/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ files:
88

99
```bash
1010
# Run all GEMM layers for MobileNet V2
11-
bazel run -c opt //bench:f32_gemm_bench -- $(cat bench/models/mobilenet_v2.txt)
11+
bazel run -c opt //bench:f32_gemm_bench -- $(cat bench/models/mobilenet_v2_gemm.txt)
1212

1313
# Run a custom Convolution shape (H W KH KW PH PW S D GCin GCout)
1414
bazel run -c opt //bench:f32_conv_hwc_bench -- 224 224 3 3 1 1 2 1 3 32
@@ -33,9 +33,9 @@ Representative shapes for various models are provided in the `models/`
3333
subdirectory.
3434

3535
### Common Models (`models/`)
36-
- **MobileNet**: `mobilenet_v1.txt`, `mobilenet_v2.txt`, `mobilenet_v3_small.txt`, `mobilenet_v3_large.txt` (and variants for `_conv.txt`, `_dwconv.txt`, `_spmm.txt`)
37-
- **ResNet**: `resnet18_gemm.txt`, `resnet50.txt`
38-
- **ShuffleNet**: `shufflenet_v1_g*.txt`, `shufflenet_v2_x*.txt`
36+
- **MobileNet**: `mobilenet_v1_gemm.txt`, `mobilenet_v2_gemm.txt`, `mobilenet_v3_small_gemm.txt`, `mobilenet_v3_large_gemm.txt` (and variants for `_conv.txt`, `_dwconv.txt`, `_spmm.txt`)
37+
- **ResNet**: `resnet18_gemm.txt`, `resnet50_gemm.txt`
38+
- **ShuffleNet**: `shufflenet_v1_g*_gemm.txt`, `shufflenet_v2_x*_gemm.txt`
3939
- **Inception**: `inception_v3_gemm.txt`
4040
- **LLM / Attention**: `llm_gemm.txt`, `attention_bgemm.txt`, `sd1x_diffusion_bgemm.txt`
4141

0 commit comments

Comments
 (0)