52ca149931
This commit extends the `ResourceLimitsAttr` to support specifying a minimal and maximal subgroup size, and extends `EntryPointABIAttr` to support specifying the requested subgroup size. This is possible now in Vulkan with the VK_EXT_subgroup_size_control extension. For OpenCL it's possible to use the `SubgroupSize` execution mode directly. Reviewed By: ThomasRaoux Differential Revision: https://reviews.llvm.org/D138962
41 lines
1.8 KiB
MLIR
41 lines
1.8 KiB
MLIR
// RUN: mlir-opt -split-input-file -spirv-lower-abi-attrs %s | FileCheck %s
|
|
|
|
module attributes {
|
|
spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Kernel, Addresses], []>, #spirv.resource_limits<>>
|
|
} {
|
|
spirv.module Physical64 OpenCL {
|
|
// CHECK-LABEL: spirv.module
|
|
// CHECK: spirv.func [[FN:@.*]]({{%.*}}: f32, {{%.*}}: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, CrossWorkgroup>
|
|
// We cannot generate SubgroupSize execution mode without necessary capability -- leave it alone.
|
|
// CHECK-SAME: #spirv.entry_point_abi<subgroup_size = 64>
|
|
// CHECK: spirv.EntryPoint "Kernel" [[FN]]
|
|
// CHECK: spirv.ExecutionMode [[FN]] "LocalSize", 32, 1, 1
|
|
spirv.func @kernel(
|
|
%arg0: f32,
|
|
%arg1: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, CrossWorkgroup>) "None"
|
|
attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [32, 1, 1], subgroup_size = 64>} {
|
|
spirv.Return
|
|
}
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes {
|
|
spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Kernel, SubgroupDispatch], []>, #spirv.resource_limits<>>
|
|
} {
|
|
spirv.module Physical64 OpenCL {
|
|
// CHECK-LABEL: spirv.module
|
|
// CHECK: spirv.func [[FN:@.*]]({{%.*}}: f32, {{%.*}}: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, CrossWorkgroup>
|
|
// CHECK: spirv.EntryPoint "Kernel" [[FN]]
|
|
// CHECK: spirv.ExecutionMode [[FN]] "LocalSize", 32, 1, 1
|
|
// CHECK: spirv.ExecutionMode [[FN]] "SubgroupSize", 64
|
|
spirv.func @kernel(
|
|
%arg0: f32,
|
|
%arg1: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, CrossWorkgroup>) "None"
|
|
attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [32, 1, 1], subgroup_size = 64>} {
|
|
spirv.Return
|
|
}
|
|
}
|
|
}
|