58df5e6d9a
This commit unifies target environment queries into a new wrapper class spirv::TargetEnv and shares across various places needing the functionality. We still create multiple instances of TargetEnv though given the parent components (type converters, passes, conversion targets) have different lifetimes. In the meantime, LowerABIAttributesPass is updated to take into consideration the target environment, which requires updates to tests to provide that. Differential Revision: https://reviews.llvm.org/D76242
37 lines
1.6 KiB
MLIR
37 lines
1.6 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]>,
|
|
{max_compute_workgroup_invocations = 128 : i32,
|
|
max_compute_workgroup_size = dense<[128, 128, 64]> : vector<3xi32>}>
|
|
} {
|
|
|
|
// 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 [4]> [0]>, StorageBuffer>
|
|
// CHECK: spv.func [[FN:@.*]]()
|
|
spv.func @kernel(
|
|
%arg0: f32
|
|
{spv.interface_var_abi = {binding = 0 : i32,
|
|
descriptor_set = 0 : i32,
|
|
storage_class = 12 : i32}},
|
|
%arg1: !spv.ptr<!spv.struct<!spv.array<12 x f32>>, StorageBuffer>
|
|
{spv.interface_var_abi = {binding = 1 : i32,
|
|
descriptor_set = 0 : i32}}) "None"
|
|
attributes {spv.entry_point_abi = {local_size = dense<[32, 1, 1]> : vector<3xi32>}} {
|
|
// CHECK: [[ARG1:%.*]] = spv._address_of [[VAR1]]
|
|
// CHECK: [[ADDRESSARG0:%.*]] = spv._address_of [[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
|