1// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %s | FileCheck %s 2 3spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> { 4 spirv.GlobalVariable @var1 : !spirv.ptr<!spirv.array<4xf32>, Input> 5 spirv.func @fmain() -> i32 "None" { 6 %0 = spirv.Constant 16 : i32 7 %1 = spirv.mlir.addressof @var1 : !spirv.ptr<!spirv.array<4xf32>, Input> 8 // CHECK: {{%.*}} = spirv.FunctionCall @f_0({{%.*}}) : (i32) -> i32 9 %3 = spirv.FunctionCall @f_0(%0) : (i32) -> i32 10 // CHECK: spirv.FunctionCall @f_1({{%.*}}, {{%.*}}) : (i32, !spirv.ptr<!spirv.array<4 x f32>, Input>) -> () 11 spirv.FunctionCall @f_1(%3, %1) : (i32, !spirv.ptr<!spirv.array<4xf32>, Input>) -> () 12 // CHECK: {{%.*}} = spirv.FunctionCall @f_2({{%.*}}) : (!spirv.ptr<!spirv.array<4 x f32>, Input>) -> !spirv.ptr<!spirv.array<4 x f32>, Input> 13 %4 = spirv.FunctionCall @f_2(%1) : (!spirv.ptr<!spirv.array<4xf32>, Input>) -> !spirv.ptr<!spirv.array<4xf32>, Input> 14 spirv.ReturnValue %3 : i32 15 } 16 spirv.func @f_0(%arg0 : i32) -> i32 "None" { 17 spirv.ReturnValue %arg0 : i32 18 } 19 spirv.func @f_1(%arg0 : i32, %arg1 : !spirv.ptr<!spirv.array<4xf32>, Input>) -> () "None" { 20 spirv.Return 21 } 22 spirv.func @f_2(%arg0 : !spirv.ptr<!spirv.array<4xf32>, Input>) -> !spirv.ptr<!spirv.array<4xf32>, Input> "None" { 23 spirv.ReturnValue %arg0 : !spirv.ptr<!spirv.array<4xf32>, Input> 24 } 25 26 spirv.func @f_loop_with_function_call(%count : i32) -> () "None" { 27 %zero = spirv.Constant 0: i32 28 %var = spirv.Variable init(%zero) : !spirv.ptr<i32, Function> 29 spirv.mlir.loop { 30 spirv.Branch ^header 31 ^header: 32 %val0 = spirv.Load "Function" %var : i32 33 %cmp = spirv.SLessThan %val0, %count : i32 34 spirv.BranchConditional %cmp, ^body, ^merge 35 ^body: 36 spirv.Branch ^continue 37 ^continue: 38 // CHECK: spirv.FunctionCall @f_inc({{%.*}}) : (!spirv.ptr<i32, Function>) -> () 39 spirv.FunctionCall @f_inc(%var) : (!spirv.ptr<i32, Function>) -> () 40 spirv.Branch ^header 41 ^merge: 42 spirv.mlir.merge 43 } 44 spirv.Return 45 } 46 spirv.func @f_inc(%arg0 : !spirv.ptr<i32, Function>) -> () "None" { 47 %one = spirv.Constant 1 : i32 48 %0 = spirv.Load "Function" %arg0 : i32 49 %1 = spirv.IAdd %0, %one : i32 50 spirv.Store "Function" %arg0, %1 : i32 51 spirv.Return 52 } 53} 54