bbffece383
Per SPIR-V validation rules, explict layout decorations are only needed for StorageBuffer, PhysicalStorageBuffer, Uniform, and PushConstant storage classes. (And even that is for Shader capabilities). So we don't need such decorations on the rest. Reviewed By: hanchung Differential Revision: https://reviews.llvm.org/D124543
31 lines
1.2 KiB
MLIR
31 lines
1.2 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect -convert-gpu-to-spirv -verify-diagnostics %s -o - | FileCheck %s
|
|
|
|
module attributes {
|
|
gpu.container_module,
|
|
spv.target_env = #spv.target_env<#spv.vce<v1.0, [Kernel, Addresses], []>, {}>
|
|
} {
|
|
gpu.module @kernels {
|
|
// CHECK-LABEL: spv.module @{{.*}} Physical64 OpenCL
|
|
// CHECK: spv.func
|
|
// CHECK-SAME: {{%.*}}: f32
|
|
// CHECK-NOT: spv.interface_var_abi
|
|
// CHECK-SAME: {{%.*}}: !spv.ptr<!spv.struct<(!spv.array<12 x f32>)>, CrossWorkgroup>
|
|
// CHECK-NOT: spv.interface_var_abi
|
|
// CHECK-SAME: spv.entry_point_abi = {local_size = dense<[32, 4, 1]> : vector<3xi32>}
|
|
gpu.func @basic_module_structure(%arg0 : f32, %arg1 : memref<12xf32, 11>) kernel
|
|
attributes {spv.entry_point_abi = {local_size = dense<[32, 4, 1]>: vector<3xi32>}} {
|
|
gpu.return
|
|
}
|
|
}
|
|
|
|
func.func @main() {
|
|
%0 = "op"() : () -> (f32)
|
|
%1 = "op"() : () -> (memref<12xf32, 11>)
|
|
%cst = arith.constant 1 : index
|
|
gpu.launch_func @kernels::@basic_module_structure
|
|
blocks in (%cst, %cst, %cst) threads in (%cst, %cst, %cst)
|
|
args(%0 : f32, %1 : memref<12xf32, 11>)
|
|
return
|
|
}
|
|
}
|