Skip to content

Commit 06ea627

Browse files
committed
Updating GEMMBenchmark constructor to use ConstantOrFunction
Builds with XNNPACK_ENABLE_ARM_SME2=ON are failing. Signed-off-by: Colm Donelan <[email protected]>
1 parent 0303481 commit 06ea627

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

bench/gemm-benchmark.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -789,7 +789,7 @@ void GEMMBenchmark(benchmark::State& state,
789789
xnn_init_f32_minmax_params_fn init_minmax_params,
790790
xnn_pack_weights_and_biases_fn pack_weights,
791791
xnn_packed_stride_weights_and_biases_fn packed_stride,
792-
size_t mr, size_t nr, size_t kr, size_t sr, size_t mr_packed,
792+
ConstantOrFunction mr, ConstantOrFunction nr, size_t kr, size_t sr, ConstantOrFunction mr_packed,
793793
uint64_t arch_flags) {
794794
if (!benchmark::utils::CheckArchFlags(state, arch_flags)) {
795795
return;

bench/gemm-benchmark.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ void GEMMBenchmark(benchmark::State& state,
9898
xnn_init_f32_minmax_params_fn init_minmax_params,
9999
xnn_pack_weights_and_biases_fn pack_weights,
100100
xnn_packed_stride_weights_and_biases_fn packed_stride,
101-
size_t mr, size_t nr, size_t kr, size_t sr, size_t mr_packed,
101+
ConstantOrFunction mr, ConstantOrFunction nr, size_t kr, size_t sr, ConstantOrFunction mr_packed,
102102
uint64_t arch_flags = 0);
103103

104104
void GEMMBenchmark(benchmark::State& state,

0 commit comments

Comments
 (0)