59 lines · plain
1// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %s | FileCheck %s2 3// RUN: %if spirv-tools %{ rm -rf %t %}4// RUN: %if spirv-tools %{ mkdir %t %}5// RUN: %if spirv-tools %{ mlir-translate --no-implicit-module --serialize-spirv --split-input-file --spirv-save-validation-files-with-prefix=%t/module %s %}6// RUN: %if spirv-tools %{ spirv-val %t %}7 8spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, VariablePointers, Linkage], [SPV_KHR_storage_buffer_storage_class, SPV_KHR_variable_pointers]> {9 spirv.GlobalVariable @var1 : !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>10 spirv.func @fmain() -> i32 "None" {11 %0 = spirv.Constant 16 : i3212 %1 = spirv.mlir.addressof @var1 : !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>13 // CHECK: {{%.*}} = spirv.FunctionCall @f_0({{%.*}}) : (i32) -> i3214 %3 = spirv.FunctionCall @f_0(%0) : (i32) -> i3215 // CHECK: spirv.FunctionCall @f_1({{%.*}}, {{%.*}}) : (i32, !spirv.ptr<!spirv.array<4 x f32>, StorageBuffer>) -> ()16 spirv.FunctionCall @f_1(%3, %1) : (i32, !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>) -> ()17 // CHECK: {{%.*}} = spirv.FunctionCall @f_2({{%.*}}) : (!spirv.ptr<!spirv.array<4 x f32>, StorageBuffer>) -> !spirv.ptr<!spirv.array<4 x f32>, StorageBuffer>18 %4 = spirv.FunctionCall @f_2(%1) : (!spirv.ptr<!spirv.array<4xf32>, StorageBuffer>) -> !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>19 spirv.ReturnValue %3 : i3220 }21 spirv.func @f_0(%arg0 : i32) -> i32 "None" {22 spirv.ReturnValue %arg0 : i3223 }24 spirv.func @f_1(%arg0 : i32, %arg1 : !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>) -> () "None" {25 spirv.Return26 }27 spirv.func @f_2(%arg0 : !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>) -> !spirv.ptr<!spirv.array<4xf32>, StorageBuffer> "None" {28 spirv.ReturnValue %arg0 : !spirv.ptr<!spirv.array<4xf32>, StorageBuffer>29 }30 31 spirv.func @f_loop_with_function_call(%count : i32) -> () "None" {32 %zero = spirv.Constant 0: i3233 %var = spirv.Variable init(%zero) : !spirv.ptr<i32, Function>34 spirv.mlir.loop {35 spirv.Branch ^header36 ^header:37 %val0 = spirv.Load "Function" %var : i3238 %cmp = spirv.SLessThan %val0, %count : i3239 spirv.BranchConditional %cmp, ^body, ^merge40 ^body:41 spirv.Branch ^continue42 ^continue:43 // CHECK: spirv.FunctionCall @f_inc({{%.*}}) : (!spirv.ptr<i32, Function>) -> ()44 spirv.FunctionCall @f_inc(%var) : (!spirv.ptr<i32, Function>) -> ()45 spirv.Branch ^header46 ^merge:47 spirv.mlir.merge48 }49 spirv.Return50 }51 spirv.func @f_inc(%arg0 : !spirv.ptr<i32, Function>) -> () "None" {52 %one = spirv.Constant 1 : i3253 %0 = spirv.Load "Function" %arg0 : i3254 %1 = spirv.IAdd %0, %one : i3255 spirv.Store "Function" %arg0, %1 : i3256 spirv.Return57 }58}59