sw/vec_iree: Fix compilation failure due to ukernel upstream API change
Change-Id: I559f4b4a423ed5850db2354719343646362add48
diff --git a/vmvx_ukernel/mmt4d_tile.c b/vmvx_ukernel/mmt4d_tile.c
index 10105b6..28d8145 100644
--- a/vmvx_ukernel/mmt4d_tile.c
+++ b/vmvx_ukernel/mmt4d_tile.c
@@ -43,8 +43,7 @@
// RVV implementation of matmul tile, i8*i8->i32 case.
static void iree_uk_mmt4d_tile_i8i8i32_rvv(
void* out_tile_untyped, const void* lhs_panel_untyped,
- const void* rhs_panel_untyped, iree_uk_int32_t K, iree_uk_uint32_t flags,
- const iree_uk_mmt4d_params_t* params) {
+ const void* rhs_panel_untyped, const iree_uk_mmt4d_params_t* params) {
iree_uk_int32_t* out_tile = out_tile_untyped;
const iree_uk_int8_t* lhs_panel = lhs_panel_untyped;
const iree_uk_int8_t* rhs_panel = rhs_panel_untyped;
@@ -52,11 +51,11 @@
iree_uk_int16_t N0 = params->N0;
iree_uk_int16_t K0 = params->K0;
// Initialize the accumulator tile.
- if (!(flags & IREE_UK_FLAG_MMT4D_ACCUMULATE)) {
+ if (!(params->flags & IREE_UK_FLAG_MMT4D_ACCUMULATE)) {
memset(out_tile, 0, M0 * N0 * sizeof(iree_uk_int32_t));
}
// Accumulation loop.
- for (iree_uk_index_t k = 0; k < K; ++k) {
+ for (iree_uk_index_t k = 0; k < params->K; ++k) {
for (iree_uk_index_t i0 = 0; i0 < M0; ++i0) {
for (iree_uk_index_t j0 = 0; j0 < N0; ++j0) {
out_tile[i0 * N0 + j0] +=
@@ -71,8 +70,7 @@
// Generic implementation of matmul tile, f32*f32->f32 case.
static void iree_uk_mmt4d_tile_f32f32f32_generic(
void* out_tile_untyped, const void* lhs_panel_untyped,
- const void* rhs_panel_untyped, iree_uk_int32_t K, iree_uk_uint32_t flags,
- const iree_uk_mmt4d_params_t* params) {
+ const void* rhs_panel_untyped, const iree_uk_mmt4d_params_t* params) {
float* out_tile = out_tile_untyped;
const float* lhs_panel = lhs_panel_untyped;
const float* rhs_panel = rhs_panel_untyped;
@@ -81,13 +79,13 @@
iree_uk_int16_t K0 = params->K0;
// Initialize the local accumulator tile.
float acc[iree_uk_mmt4d_tile_generic_max_bytes / sizeof(*out_tile)];
- if (flags & IREE_UK_FLAG_MMT4D_ACCUMULATE) {
+ if (params->flags & IREE_UK_FLAG_MMT4D_ACCUMULATE) {
for (int i = 0; i < M0 * N0; ++i) acc[i] = out_tile[i];
} else {
for (int i = 0; i < M0 * N0; ++i) acc[i] = 0;
}
// Accumulation loop.
- for (iree_uk_index_t k = 0; k < K; ++k) {
+ for (iree_uk_index_t k = 0; k < params->K; ++k) {
for (iree_uk_index_t i0 = 0; i0 < M0; ++i0) {
for (iree_uk_index_t j0 = 0; j0 < N0; ++j0) {
for (iree_uk_index_t k0 = 0; k0 < K0; ++k0) {
