1// RUN: mlir-opt -test-convert-to-spirv="run-signature-conversion=false run-vector-unrolling=false" -split-input-file %s | FileCheck %s 2 3// CHECK-LABEL: @combined 4// CHECK: %[[C0_F32:.*]] = spirv.Constant 0.000000e+00 : f32 5// CHECK: %[[C1_F32:.*]] = spirv.Constant 1.000000e+00 : f32 6// CHECK: %[[C0_I32:.*]] = spirv.Constant 0 : i32 7// CHECK: %[[C4_I32:.*]] = spirv.Constant 4 : i32 8// CHECK: %[[C0_I32_0:.*]] = spirv.Constant 0 : i32 9// CHECK: %[[C4_I32_0:.*]] = spirv.Constant 4 : i32 10// CHECK: %[[C1_I32:.*]] = spirv.Constant 1 : i32 11// 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]] : i32 17// CHECK: spirv.BranchConditional %[[SLESSTHAN]], ^[[BODY:.*]], ^[[MERGE:.*]] 18// CHECK: ^[[BODY]]: 19// CHECK: %[[FADD:.*]] = spirv.FAdd %[[INDVAR_1]], %[[C1_F32]] : f32 20// CHECK: %[[INSERT:.*]] = spirv.CompositeInsert %[[FADD]], %[[VEC]][0 : i32] : f32 into vector<4xf32> 21// CHECK: spirv.Store "Function" %[[VARIABLE]], %[[FADD]] : f32 22// CHECK: %[[IADD:.*]] = spirv.IAdd %[[INDVAR_0]], %[[C1_I32]] : i32 23// CHECK: spirv.Branch ^[[HEADER]](%[[IADD]], %[[FADD]] : i32, f32) 24// CHECK: ^[[MERGE]]: 25// CHECK: spirv.mlir.merge 26// CHECK: } 27// CHECK: %[[LOAD:.*]] = spirv.Load "Function" %[[VARIABLE]] : f32 28// CHECK: %[[UNDEF:.*]] = spirv.Undef : f32 29// CHECK: spirv.ReturnValue %[[UNDEF]] : f32 30func.func @combined() -> f32 { 31 %c0_f32 = arith.constant 0.0 : f32 32 %c1_f32 = arith.constant 1.0 : f32 33 %c0_i32 = arith.constant 0 : i32 34 %c4_i32 = arith.constant 4 : i32 35 %lb = index.casts %c0_i32 : i32 to index 36 %ub = index.casts %c4_i32 : i32 to index 37 %step = arith.constant 1 : index 38 %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 : f32 42 vector.insert %sum_next, %buf[0] : f32 into vector<4xf32> 43 scf.yield %sum_next : f32 44 } 45 %ret = ub.poison : f32 46 return %ret : f32 47} 48