// File: binding.cpp
#include "gemm.h"

namespace {
constexpr int64_t kN = 2048;
constexpr int64_t kK = 1024;
} // namespace

extern "C" {
// inputs: [0]=A bf16 [M,K], [1]=B bf16 [N,K]; var_axes: [0]=M
int armbench_entry_gemm(const void* const* inputs, void* output,
                        const int64_t* var_axes)
{
    return armbench_llamacpp_gemm(inputs[0], inputs[1],
                                  reinterpret_cast<float*>(output),
                                  var_axes[0], kN, kK, /*is_q8=*/false);
}
} // extern "C"