Function prompt_flat

src/models/heartmula/generator.cpp:24–26

};

size_t prompt_flat(size_t batch, size_t row, size_t lane, size_t steps, size_t lanes) {
    return (batch * steps + row) * lanes + lane;
}

int32_t sample_topk_row(
    const float * logits,