41 lines
1.8 KiB
MLIR
41 lines
1.8 KiB
MLIR
// RUN: mlir-opt --split-input-file --spirv-lower-abi-attrs --verify-diagnostics %s \
|
|
// RUN: | FileCheck %s
|
|
|
|
module attributes {
|
|
spirv.target_env = #spirv.target_env<
|
|
#spirv.vce<v1.0, [Shader], [SPV_KHR_storage_buffer_storage_class]>, #spirv.resource_limits<>>
|
|
} {
|
|
|
|
// CHECK-LABEL: spirv.module
|
|
spirv.module Logical GLSL450 {
|
|
// CHECK-DAG: spirv.GlobalVariable [[VAR0:@.*]] bind(0, 0) : !spirv.ptr<!spirv.struct<(f32 [0])>, StorageBuffer>
|
|
// CHECK-DAG: spirv.GlobalVariable [[VAR1:@.*]] bind(0, 1) : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>
|
|
// CHECK: spirv.func [[FN:@.*]]()
|
|
// We cannot generate SubgroupSize execution mode for Shader capability -- leave it alone.
|
|
// CHECK-SAME: #spirv.entry_point_abi<subgroup_size = 64>
|
|
spirv.func @kernel(
|
|
%arg0: f32
|
|
{spirv.interface_var_abi = #spirv.interface_var_abi<(0, 0), StorageBuffer>},
|
|
%arg1: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, StorageBuffer>
|
|
{spirv.interface_var_abi = #spirv.interface_var_abi<(0, 1)>}) "None"
|
|
attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [32, 1, 1], subgroup_size = 64>} {
|
|
// CHECK: [[ARG1:%.*]] = spirv.mlir.addressof [[VAR1]]
|
|
// CHECK: [[ADDRESSARG0:%.*]] = spirv.mlir.addressof [[VAR0]]
|
|
// CHECK: [[CONST0:%.*]] = spirv.Constant 0 : i32
|
|
// CHECK: [[ARG0PTR:%.*]] = spirv.AccessChain [[ADDRESSARG0]]{{\[}}[[CONST0]]
|
|
// CHECK: [[ARG0:%.*]] = spirv.Load "StorageBuffer" [[ARG0PTR]]
|
|
// CHECK: spirv.Return
|
|
spirv.Return
|
|
}
|
|
// CHECK: spirv.EntryPoint "GLCompute" [[FN]]
|
|
// CHECK: spirv.ExecutionMode [[FN]] "LocalSize", 32, 1, 1
|
|
} // end spirv.module
|
|
|
|
} // end module
|
|
|
|
// -----
|
|
|
|
module {
|
|
// expected-error@+1 {{'spirv.module' op missing SPIR-V target env attribute}}
|
|
spirv.module Logical GLSL450 {}
|
|
} // end module
|