1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
// 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
}
}
}
|
