a062a3ed7f
Implement a pass to convert gpu.launch_func op into a sequence of Vulkan runtime calls. The Vulkan runtime API surface is huge so currently we don't expose separate external functions in IR for each of them, instead we expose a few external functions to wrapper libraries which manages Vulkan runtime. Differential Revision: https://reviews.llvm.org/D74549
46 lines
2.4 KiB
MLIR
46 lines
2.4 KiB
MLIR
// RUN: mlir-opt %s -launch-func-to-vulkan | FileCheck %s
|
|
|
|
// CHECK: llvm.mlir.global internal constant @kernel_spv_entry_point_name
|
|
// CHECK: llvm.mlir.global internal constant @SPIRV_BIN
|
|
// CHECK: %[[addressof_SPIRV_BIN:.*]] = llvm.mlir.addressof @SPIRV_BIN
|
|
// CHECK: %[[SPIRV_BIN_ptr:.*]] = llvm.getelementptr %[[addressof_SPIRV_BIN]]
|
|
// CHECK: %[[SPIRV_BIN_size:.*]] = llvm.mlir.constant
|
|
// CHECK: llvm.call @setBinaryShader(%[[SPIRV_BIN_ptr]], %[[SPIRV_BIN_size]]) : (!llvm<"i8*">, !llvm.i32) -> !llvm.void
|
|
// CHECK: %[[addressof_entry_point:.*]] = llvm.mlir.addressof @kernel_spv_entry_point_name
|
|
// CHECK: %[[entry_point_ptr:.*]] = llvm.getelementptr %[[addressof_entry_point]]
|
|
// CHECK: llvm.call @setEntryPoint(%[[entry_point_ptr]]) : (!llvm<"i8*">) -> !llvm.void
|
|
// CHECK: %[[Workgroup_X:.*]] = llvm.mlir.constant
|
|
// CHECK: %[[Workgroup_Y:.*]] = llvm.mlir.constant
|
|
// CHECK: %[[Workgroup_Z:.*]] = llvm.mlir.constant
|
|
// CHECK: llvm.call @setNumWorkGroups(%[[Workgroup_X]], %[[Workgroup_Y]], %[[Workgroup_Z]]) : (!llvm.i32, !llvm.i32, !llvm.i32) -> !llvm.void
|
|
// CHECK: llvm.call @runOnVulkan() : () -> !llvm.void
|
|
|
|
module attributes {gpu.container_module} {
|
|
spv.module "Logical" "GLSL450" {
|
|
spv.globalVariable @kernel_arg_0 bind(0, 0) : !spv.ptr<!spv.struct<f32 [0]>, StorageBuffer>
|
|
spv.globalVariable @kernel_arg_1 bind(0, 1) : !spv.ptr<!spv.struct<!spv.array<12 x f32 [4]> [0]>, StorageBuffer>
|
|
spv.func @kernel() "None" attributes {workgroup_attributions = 0 : i64} {
|
|
%0 = spv._address_of @kernel_arg_1 : !spv.ptr<!spv.struct<!spv.array<12 x f32 [4]> [0]>, StorageBuffer>
|
|
%1 = spv._address_of @kernel_arg_0 : !spv.ptr<!spv.struct<f32 [0]>, StorageBuffer>
|
|
%2 = spv.constant 0 : i32
|
|
%3 = spv.AccessChain %1[%2] : !spv.ptr<!spv.struct<f32 [0]>, StorageBuffer>
|
|
%4 = spv.Load "StorageBuffer" %3 : f32
|
|
spv.Return
|
|
}
|
|
spv.EntryPoint "GLCompute" @kernel
|
|
spv.ExecutionMode @kernel "LocalSize", 1, 1, 1
|
|
} attributes {capabilities = ["Shader"], extensions = ["SPV_KHR_storage_buffer_storage_class"]}
|
|
gpu.module @kernels {
|
|
gpu.func @kernel(%arg0: f32, %arg1: memref<12xf32>) kernel {
|
|
gpu.return
|
|
}
|
|
}
|
|
func @foo() {
|
|
%0 = "op"() : () -> f32
|
|
%1 = "op"() : () -> memref<12xf32>
|
|
%c1 = constant 1 : index
|
|
"gpu.launch_func"(%c1, %c1, %c1, %c1, %c1, %c1, %0, %1) {kernel = "kernel", kernel_module = @kernels} : (index, index, index, index, index, index, f32, memref<12xf32>) -> ()
|
|
return
|
|
}
|
|
}
|