mlir/test/Conversion/GPUToVulkan/lower-gpu-launch-vulkan-launch.mlir

   1 // RUN: mlir-opt %s -convert-gpu-launch-to-vulkan-launch | FileCheck %s
   2
   3 // CHECK: %[[resource:.*]] = memref.alloc() : memref<12xf32>
   4 // CHECK: %[[index:.*]] = arith.constant 1 : index
   5 // CHECK: call @vulkanLaunch(%[[index]], %[[index]], %[[index]], %[[resource]]) {spirv_blob = "{{.*}}", spirv_element_types = [f32], spirv_entry_point = "kernel"}
   6
   7 module attributes {gpu.container_module} {
   8   spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], [SPV_KHR_storage_buffer_storage_class]> {
   9     spirv.GlobalVariable @kernel_arg_0 bind(0, 0) : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>
  10     spirv.func @kernel() "None" attributes {workgroup_attributions = 0 : i64} {
  11       %0 = spirv.mlir.addressof @kernel_arg_0 : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>
  12       %2 = spirv.Constant 0 : i32
  13       %3 = spirv.mlir.addressof @kernel_arg_0 : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>
  14       %4 = spirv.AccessChain %0[%2, %2] : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>, i32, i32 -> !spirv.ptr<f32, StorageBuffer>
  15       %5 = spirv.Load "StorageBuffer" %4 : f32
  16       spirv.Return
  17     }
  18     spirv.EntryPoint "GLCompute" @kernel
  19     spirv.ExecutionMode @kernel "LocalSize", 1, 1, 1
  20   }
  21   gpu.module @kernels {
  22     gpu.func @kernel(%arg0: memref<12xf32>) kernel {
  23       gpu.return
  24     }
  25   }
  26   func.func @foo() {
  27     %0 = memref.alloc() : memref<12xf32>
  28     %c1 = arith.constant 1 : index
  29     gpu.launch_func @kernels::@kernel
  30         blocks in(%c1, %c1, %c1)
  31         threads in(%c1, %c1, %c1)
  32         args(%0 : memref<12xf32>)
  33     return
  34   }
  35 }