48 lines · plain
1// RUN: mlir-opt -test-convert-to-spirv="run-signature-conversion=false run-vector-unrolling=false" -split-input-file %s | FileCheck %s2 3// CHECK-LABEL: @combined4// CHECK: %[[C0_F32:.*]] = spirv.Constant 0.000000e+00 : f325// CHECK: %[[C1_F32:.*]] = spirv.Constant 1.000000e+00 : f326// CHECK: %[[C0_I32:.*]] = spirv.Constant 0 : i327// CHECK: %[[C4_I32:.*]] = spirv.Constant 4 : i328// CHECK: %[[C0_I32_0:.*]] = spirv.Constant 0 : i329// CHECK: %[[C4_I32_0:.*]] = spirv.Constant 4 : i3210// CHECK: %[[C1_I32:.*]] = spirv.Constant 1 : i3211// CHECK: %[[VEC:.*]] = spirv.Constant dense<1.000000e+00> : vector<4xf32>12// CHECK: %[[VARIABLE:.*]] = spirv.Variable : !spirv.ptr<f32, Function>13// CHECK: spirv.mlir.loop {14// CHECK: spirv.Branch ^[[HEADER:.*]](%[[C0_I32_0]], %[[C0_F32]] : i32, f32)15// CHECK: ^[[HEADER]](%[[INDVAR_0:.*]]: i32, %[[INDVAR_1:.*]]: f32):16// CHECK: %[[SLESSTHAN:.*]] = spirv.SLessThan %[[INDVAR_0]], %[[C4_I32_0]] : i3217// CHECK: spirv.BranchConditional %[[SLESSTHAN]], ^[[BODY:.*]], ^[[MERGE:.*]]18// CHECK: ^[[BODY]]:19// CHECK: %[[FADD:.*]] = spirv.FAdd %[[INDVAR_1]], %[[C1_F32]] : f3220// CHECK: %[[INSERT:.*]] = spirv.CompositeInsert %[[FADD]], %[[VEC]][0 : i32] : f32 into vector<4xf32>21// CHECK: spirv.Store "Function" %[[VARIABLE]], %[[FADD]] : f3222// CHECK: %[[IADD:.*]] = spirv.IAdd %[[INDVAR_0]], %[[C1_I32]] : i3223// CHECK: spirv.Branch ^[[HEADER]](%[[IADD]], %[[FADD]] : i32, f32)24// CHECK: ^[[MERGE]]:25// CHECK: spirv.mlir.merge26// CHECK: }27// CHECK: %[[LOAD:.*]] = spirv.Load "Function" %[[VARIABLE]] : f3228// CHECK: %[[UNDEF:.*]] = spirv.Undef : f3229// CHECK: spirv.ReturnValue %[[UNDEF]] : f3230func.func @combined() -> f32 {31 %c0_f32 = arith.constant 0.0 : f3232 %c1_f32 = arith.constant 1.0 : f3233 %c0_i32 = arith.constant 0 : i3234 %c4_i32 = arith.constant 4 : i3235 %lb = index.casts %c0_i32 : i32 to index36 %ub = index.casts %c4_i32 : i32 to index37 %step = arith.constant 1 : index38 %buf = vector.broadcast %c1_f32 : f32 to vector<4xf32>39 scf.for %iv = %lb to %ub step %step iter_args(%sum_iter = %c0_f32) -> f32 {40 %t = vector.extract %buf[0] : f32 from vector<4xf32>41 %sum_next = arith.addf %sum_iter, %t : f3242 vector.insert %sum_next, %buf[0] : f32 into vector<4xf32>43 scf.yield %sum_next : f3244 }45 %ret = ub.poison : f3246 return %ret : f3247}48