
This change adds a folder for the VecExtractOp Issue https://github.com/llvm/llvm-project/issues/136487
36 lines
1.7 KiB
Plaintext
36 lines
1.7 KiB
Plaintext
// 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>
|
|
}
|
|
|
|
|