Skip to content

Commit

Permalink
Fix bench_sqnbitgemm.cpp benchmark argument name list. (#20858)
Browse files Browse the repository at this point in the history
Add the "HasBias" argument to the ArgNames() call so it matches with the ArgsProduct() call.
  • Loading branch information
edgchen1 authored May 31, 2024
1 parent b02d5e6 commit 00589f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ void SQNBITGEMM(benchmark::State& state) {
}

static void SQNBitGemmArgs(benchmark::internal::Benchmark* b) {
b->ArgNames({"BlkLen", "M", "N", "K", "Threads", "Symmetric", "ComputeType"});
b->ArgNames({"BlkLen", "M", "N", "K", "Threads", "Symmetric", "HasBias", "ComputeType"});

b->ArgsProduct({
{16, 32, 64, 128, 256}, // BlkLen
Expand Down

0 comments on commit 00589f5

Please sign in to comment.