37 lines · plain
1// RUN: mlir-opt %s -test-lower-to-llvm | \2// RUN: mlir-runner -e entry -entry-point-result=void \3// RUN: -shared-libs=%mlir_c_runner_utils | \4// RUN: FileCheck %s5 6func.func @entry() {7 %f1 = arith.constant 1.0: f328 %f2 = arith.constant 2.0: f329 %f3 = arith.constant 3.0: f3210 %f4 = arith.constant 4.0: f3211 %v1 = vector.broadcast %f1 : f32 to vector<4xf32>12 %v2 = vector.broadcast %f2 : f32 to vector<3xf32>13 %v3 = vector.broadcast %f3 : f32 to vector<4x4xf32>14 %v4 = vector.broadcast %f4 : f32 to vector<1xf32>15 16 %s1 = vector.insert_strided_slice %v1, %v3 {offsets = [2, 0], strides = [1]} : vector<4xf32> into vector<4x4xf32>17 %s2 = vector.insert_strided_slice %v2, %s1 {offsets = [1, 1], strides = [1]} : vector<3xf32> into vector<4x4xf32>18 %s3 = vector.insert_strided_slice %v2, %s2 {offsets = [0, 0], strides = [1]} : vector<3xf32> into vector<4x4xf32>19 %s4 = vector.insert_strided_slice %v4, %s3 {offsets = [3, 3], strides = [1]} : vector<1xf32> into vector<4x4xf32>20 21 vector.print %v3 : vector<4x4xf32>22 vector.print %s1 : vector<4x4xf32>23 vector.print %s2 : vector<4x4xf32>24 vector.print %s3 : vector<4x4xf32>25 vector.print %s4 : vector<4x4xf32>26 //27 // insert strided slice:28 //29 // CHECK: ( ( 3, 3, 3, 3 ), ( 3, 3, 3, 3 ), ( 3, 3, 3, 3 ), ( 3, 3, 3, 3 ) )30 // CHECK: ( ( 3, 3, 3, 3 ), ( 3, 3, 3, 3 ), ( 1, 1, 1, 1 ), ( 3, 3, 3, 3 ) )31 // CHECK: ( ( 3, 3, 3, 3 ), ( 3, 2, 2, 2 ), ( 1, 1, 1, 1 ), ( 3, 3, 3, 3 ) )32 // CHECK: ( ( 2, 2, 2, 3 ), ( 3, 2, 2, 2 ), ( 1, 1, 1, 1 ), ( 3, 3, 3, 3 ) )33 // CHECK: ( ( 2, 2, 2, 3 ), ( 3, 2, 2, 2 ), ( 1, 1, 1, 1 ), ( 3, 3, 3, 4 ) )34 35 return36}37