a31ff0af9b
Depends on D127370 Reviewed By: antiagainst Differential Revision: https://reviews.llvm.org/D127373
32 lines
1.4 KiB
MLIR
32 lines
1.4 KiB
MLIR
// 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]>, #spv.resource_limits<>>
|
|
} {
|
|
|
|
// 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 = #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
|