| // RUN: mlir-opt -spirv-lower-abi-attrs -verify-diagnostics %s -o - | FileCheck %s |
| |
| module attributes { |
| spv.target_env = #spv.target_env< |
| #spv.vce<v1.0, [Shader], [SPV_KHR_storage_buffer_storage_class]>, {}> |
| } { |
| |
| // CHECK-LABEL: spv.module |
| spv.module Logical GLSL450 { |
| // CHECK-DAG: spv.GlobalVariable [[VAR0:@.*]] bind(0, 0) : !spv.ptr<!spv.struct<(f32 [0])>, StorageBuffer> |
| // CHECK-DAG: spv.GlobalVariable [[VAR1:@.*]] bind(0, 1) : !spv.ptr<!spv.struct<(!spv.array<12 x f32, stride=4> [0])>, StorageBuffer> |
| // CHECK: spv.func [[FN:@.*]]() |
| spv.func @kernel( |
| %arg0: f32 |
| {spv.interface_var_abi = #spv.interface_var_abi<(0, 0), StorageBuffer>}, |
| %arg1: !spv.ptr<!spv.struct<(!spv.array<12 x f32>)>, StorageBuffer> |
| {spv.interface_var_abi = #spv.interface_var_abi<(0, 1)>}) "None" |
| attributes {spv.entry_point_abi = {local_size = dense<[32, 1, 1]> : vector<3xi32>}} { |
| // CHECK: [[ARG1:%.*]] = spv.mlir.addressof [[VAR1]] |
| // CHECK: [[ADDRESSARG0:%.*]] = spv.mlir.addressof [[VAR0]] |
| // CHECK: [[CONST0:%.*]] = spv.Constant 0 : i32 |
| // CHECK: [[ARG0PTR:%.*]] = spv.AccessChain [[ADDRESSARG0]]{{\[}}[[CONST0]] |
| // CHECK: [[ARG0:%.*]] = spv.Load "StorageBuffer" [[ARG0PTR]] |
| // CHECK: spv.Return |
| spv.Return |
| } |
| // CHECK: spv.EntryPoint "GLCompute" [[FN]] |
| // CHECK: spv.ExecutionMode [[FN]] "LocalSize", 32, 1, 1 |
| } // end spv.module |
| |
| } // end module |