llvm/mlir/test/Dialect/SPIRV/Transforms/abi-interface-opencl.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
    }
  }
}