// File: mla_prefill.cpp
// Binding harness: forwards straight to inner_mla_prefill (no derived dims needed).
// ABI: armbench_entry_mla_prefill(Q*, output*, K*, V*, M, S)
#include "mla_prefill.h"
using namespace mla_prefill_def;

extern "C" int armbench_entry_mla_prefill(const uint16_t* Q, float* output,
                                          const uint16_t* K, const uint16_t* V,
                                          int M, int S)
{
    inner_mla_prefill(Q, output, K, V, M, S);
    return 0;
}