119 lines · plain
1// RUN: mlir-opt -int-range-optimizations -canonicalize %s | FileCheck %s2 3 4// CHECK-LABEL: func @constant_vec5// CHECK: test.reflect_bounds {smax = 7 : index, smin = 0 : index, umax = 7 : index, umin = 0 : index}6func.func @constant_vec() -> vector<8xindex> {7 %0 = arith.constant dense<[0, 1, 2, 3, 4, 5, 6, 7]> : vector<8xindex>8 %1 = test.reflect_bounds %0 : vector<8xindex>9 func.return %1 : vector<8xindex>10}11 12// CHECK-LABEL: func @constant_splat13// CHECK: test.reflect_bounds {smax = 3 : si32, smin = 3 : si32, umax = 3 : ui32, umin = 3 : ui32}14func.func @constant_splat() -> vector<8xi32> {15 %0 = arith.constant dense<3> : vector<8xi32>16 %1 = test.reflect_bounds %0 : vector<8xi32>17 func.return %1 : vector<8xi32>18}19 20// CHECK-LABEL: func @float_constant_splat21// Don't crash on splat floats.22func.func @float_constant_splat() -> vector<8xf32> {23 %0 = arith.constant dense<3.0> : vector<8xf32>24 func.return %0: vector<8xf32>25}26 27// CHECK-LABEL: func @vector_splat28// CHECK: test.reflect_bounds {smax = 5 : index, smin = 4 : index, umax = 5 : index, umin = 4 : index}29func.func @vector_splat() -> vector<4xindex> {30 %0 = test.with_bounds { umin = 4 : index, umax = 5 : index, smin = 4 : index, smax = 5 : index } : index31 %1 = vector.broadcast %0 : index to vector<4xindex>32 %2 = test.reflect_bounds %1 : vector<4xindex>33 func.return %2 : vector<4xindex>34}35 36// CHECK-LABEL: func @vector_broadcast37// CHECK: test.reflect_bounds {smax = 5 : index, smin = 4 : index, umax = 5 : index, umin = 4 : index}38func.func @vector_broadcast() -> vector<4x16xindex> {39 %0 = test.with_bounds { umin = 4 : index, umax = 5 : index, smin = 4 : index, smax = 5 : index } : vector<16xindex>40 %1 = vector.broadcast %0 : vector<16xindex> to vector<4x16xindex>41 %2 = test.reflect_bounds %1 : vector<4x16xindex>42 func.return %2 : vector<4x16xindex>43}44 45// CHECK-LABEL: func @vector_shape_cast46// CHECK: test.reflect_bounds {smax = 5 : index, smin = 4 : index, umax = 5 : index, umin = 4 : index}47func.func @vector_shape_cast() -> vector<4x4xindex> {48 %0 = test.with_bounds { umin = 4 : index, umax = 5 : index, smin = 4 : index, smax = 5 : index } : vector<16xindex>49 %1 = vector.shape_cast %0 : vector<16xindex> to vector<4x4xindex>50 %2 = test.reflect_bounds %1 : vector<4x4xindex>51 func.return %2 : vector<4x4xindex>52}53 54// CHECK-LABEL: func @vector_transpose55// CHECK: test.reflect_bounds {smax = 8 : index, smin = 7 : index, umax = 8 : index, umin = 7 : index}56func.func @vector_transpose() -> vector<2x4xindex> {57 %0 = test.with_bounds { smax = 8 : index, smin = 7 : index, umax = 8 : index, umin = 7 : index } : vector<4x2xindex>58 %1 = vector.transpose %0, [1, 0] : vector<4x2xindex> to vector<2x4xindex>59 %2 = test.reflect_bounds %1 : vector<2x4xindex>60 func.return %2 : vector<2x4xindex>61}62 63// CHECK-LABEL: func @vector_extract64// CHECK: test.reflect_bounds {smax = 6 : index, smin = 5 : index, umax = 6 : index, umin = 5 : index}65func.func @vector_extract() -> index {66 %0 = test.with_bounds { umin = 5 : index, umax = 6 : index, smin = 5 : index, smax = 6 : index } : vector<4xindex>67 %1 = vector.extract %0[0] : index from vector<4xindex>68 %2 = test.reflect_bounds %1 : index69 func.return %2 : index70}71 72// CHECK-LABEL: func @vector_add73// CHECK: test.reflect_bounds {smax = 12 : index, smin = 10 : index, umax = 12 : index, umin = 10 : index}74func.func @vector_add() -> vector<4xindex> {75 %0 = test.with_bounds { umin = 4 : index, umax = 5 : index, smin = 4 : index, smax = 5 : index } : vector<4xindex>76 %1 = test.with_bounds { umin = 6 : index, umax = 7 : index, smin = 6 : index, smax = 7 : index } : vector<4xindex>77 %2 = arith.addi %0, %1 : vector<4xindex>78 %3 = test.reflect_bounds %2 : vector<4xindex>79 func.return %3 : vector<4xindex>80}81 82// CHECK-LABEL: func @vector_insert83// CHECK: test.reflect_bounds {smax = 8 : index, smin = 5 : index, umax = 8 : index, umin = 5 : index}84func.func @vector_insert() -> vector<4xindex> {85 %0 = test.with_bounds { umin = 5 : index, umax = 7 : index, smin = 5 : index, smax = 7 : index } : vector<4xindex>86 %1 = test.with_bounds { umin = 6 : index, umax = 8 : index, smin = 6 : index, smax = 8 : index } : index87 %2 = vector.insert %1, %0[0] : index into vector<4xindex>88 %3 = test.reflect_bounds %2 : vector<4xindex>89 func.return %3 : vector<4xindex>90}91 92// CHECK-LABEL: func @test_loaded_vector_extract93// No bounds94// CHECK: test.reflect_bounds {smax = 2147483647 : si32, smin = -2147483648 : si32, umax = 4294967295 : ui32, umin = 0 : ui32} %{{.*}} : i3295func.func @test_loaded_vector_extract(%memref : memref<16xi32>) -> i32 {96 %c0 = arith.constant 0 : index97 %v = vector.load %memref[%c0] : memref<16xi32>, vector<4xi32>98 %e = vector.extract %v[0] : i32 from vector<4xi32>99 %bounds = test.reflect_bounds %e : i32100 func.return %bounds : i32101}102 103// CHECK-LABEL: func @test_vector_extsi104// CHECK: test.reflect_bounds {smax = 5 : si32, smin = 1 : si32, umax = 5 : ui32, umin = 1 : ui32}105func.func @test_vector_extsi() -> vector<2xi32> {106 %0 = test.with_bounds {smax = 5 : si8, smin = 1 : si8, umax = 5 : ui8, umin = 1 : ui8 } : vector<2xi8>107 %1 = arith.extsi %0 : vector<2xi8> to vector<2xi32>108 %2 = test.reflect_bounds %1 : vector<2xi32>109 func.return %2 : vector<2xi32>110}111 112// CHECK-LABEL: func @vector_step113// CHECK: test.reflect_bounds {smax = 7 : index, smin = 0 : index, umax = 7 : index, umin = 0 : index}114func.func @vector_step() -> vector<8xindex> {115 %0 = vector.step : vector<8xindex>116 %1 = test.reflect_bounds %0 : vector<8xindex>117 func.return %1 : vector<8xindex>118}119