/* { dg-do compile } */
/* { dg-options "-march=rv32gcv -mabi=ilp32 -fno-tree-vectorize -fno-schedule-insns -fno-schedule-insns2" } */

#include "riscv_vector.h"

void f(void *base, void *out, void *mask_in, 
size_t m, size_t n, size_t a, size_t b) {
  
  size_t vl = 345;
  for (size_t i = 0; i < m; i++) {
    if (i % 2 == 0) {
      for (size_t j = 0; j < n; j++){
        if (j % 2 == 0) {
          for (size_t k = 0; k < n; k++) {
            for (size_t i_a = 0; i_a < a; i_a++){
              for (size_t i_b = 0; i_b < b; i_b++){
                vint8mf8_t v0 = __riscv_vle8_v_i8mf8(base + i + 500 + k + j + i_a + i_b, vl);
                vint8mf8_t v1 = __riscv_vle8_v_i8mf8_tu(v0, base + i + 600 + k + j + i_a + i_b, vl);
                __riscv_vse8_v_i8mf8 (out + i + 600 + j + k + i_a + i_b, v1, vl);
              }
            }
          }
        } else {
          vint8mf8_t v0 = __riscv_vle8_v_i8mf8(base + i + 300 + j, vl);
          __riscv_vse8_v_i8mf8 (out + i + 300, v0, vl);
        }
      }
    } else {
      for (size_t j = 0; j < vl; j++){
        if (j % 2 == 0) {
          vint8mf8_t v0 = __riscv_vle8_v_i8mf8(base + i + 200 + j, vl);
          __riscv_vse8_v_i8mf8 (out + i + 200, v0, vl);
        } else {
          vint8mf8_t v0 = __riscv_vle8_v_i8mf8(base + i + 700, vl);
          __riscv_vse8_v_i8mf8 (out + i + 800, v0, vl);
        }
      }
    }
  }
}

/* { dg-final { scan-assembler-times {vsetvli\s+zero,\s*[a-x0-9]+,\s*e8,\s*mf8,\s*tu,\s*m[au]} 1 { target { no-opts "-O0" no-opts "-g" no-opts "-funroll-loops" } } } } */
/* { dg-final { scan-assembler-times {vsetvli} 1 { target { no-opts "-O0" no-opts "-g" no-opts "-funroll-loops" } } } } */