// RUN: cir-opt %s -cir-canonicalize -o - | FileCheck %s

!s32i = !cir.int<s, 32>

module {
  cir.func @fold_extract_vector_op_test() {
    %init = cir.alloca !s32i, !cir.ptr<!s32i>, ["e", init]
    %const_vec = cir.const #cir.const_vector<[#cir.int<1> : !s32i, #cir.int<2> : !s32i, #cir.int<3> : !s32i, #cir.int<4> : !s32i]> : !cir.vector<4 x !s32i>
    %index = cir.const #cir.int<1> : !s32i
    %ele = cir.vec.extract %const_vec[%index : !s32i] : !cir.vector<4 x !s32i>
    cir.store %ele, %init : !s32i, !cir.ptr<!s32i>
    cir.return
  }

  // CHECK: %[[INIT:.*]] = cir.alloca !s32i, !cir.ptr<!s32i>, ["e", init]
  // CHECK: %[[VALUE:.*]] = cir.const #cir.int<2> : !s32i
  // CHECK: cir.store %[[VALUE]], %[[INIT]] : !s32i, !cir.ptr<!s32i>

  cir.func @fold_extract_vector_op_index_out_of_bounds_test() {
    %init = cir.alloca !s32i, !cir.ptr<!s32i>, ["e", init]
    %const_vec = cir.const #cir.const_vector<[#cir.int<1> : !s32i, #cir.int<2> : !s32i, #cir.int<3> : !s32i, #cir.int<4> : !s32i]> : !cir.vector<4 x !s32i>
    %index = cir.const #cir.int<9> : !s32i
    %ele = cir.vec.extract %const_vec[%index : !s32i] : !cir.vector<4 x !s32i>
    cir.store %ele, %init : !s32i, !cir.ptr<!s32i>
    cir.return
  }

  // CHECK: %[[INIT:.*]] = cir.alloca !s32i, !cir.ptr<!s32i>, ["e", init]
  // CHECK: %[[CONST_VEC:.*]] = cir.const #cir.const_vector<[#cir.int<1> : !s32i, #cir.int<2> : !s32i, #cir.int<3> : !s32i, #cir.int<4> : !s32i]> : !cir.vector<4 x !s32i>
  // CHECK: %[[INDEX:.*]] = cir.const #cir.int<9> : !s32i
  // CHECK: %[[ELE:.*]] = cir.vec.extract %[[CONST_VEC]][%[[INDEX]] : !s32i] : !cir.vector<4 x !s32i>
  // CHECK: cir.store %[[ELE]], %[[INIT]] : !s32i, !cir.ptr<!s32i>
}