llvm/test/CodeGen/SPIRV/EnqueueEmptyKernel.ll

   1 ; RUN: llc -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV
   2
   3 ;; This test checks that Invoke parameter of OpEnueueKernel instruction meet the
   4 ;; following specification requirements in case of enqueueing empty block:
   5 ;; "Invoke must be an OpFunction whose OpTypeFunction operand has:
   6 ;; - Result Type must be OpTypeVoid.
   7 ;; - The first parameter must have a type of OpTypePointer to an 8-bit OpTypeInt.
   8 ;; - An optional list of parameters, each of which must have a type of OpTypePointer to the Workgroup Storage Class.
   9 ;; ... "
  10 ;; __kernel void test_enqueue_empty() {
  11 ;;   enqueue_kernel(get_default_queue(),
  12 ;;                  CLK_ENQUEUE_FLAGS_WAIT_KERNEL,
  13 ;;                  ndrange_1D(1),
  14 ;;                  0, NULL, NULL,
  15 ;;                  ^(){});
  16 ;; }
  17
  18 %struct.ndrange_t = type { i32, [3 x i64], [3 x i64], [3 x i64] }
  19 %opencl.queue_t = type opaque
  20 %opencl.clk_event_t = type opaque
  21
  22 @__block_literal_global = internal addrspace(1) constant { i32, i32 } { i32 8, i32 4 }, align 4
  23
  24 ; CHECK-SPIRV: OpName %[[#Block:]] "__block_literal_global"
  25 ; CHECK-SPIRV: %[[#Void:]] = OpTypeVoid
  26 ; CHECK-SPIRV: %[[#Int8:]] = OpTypeInt 8
  27 ; CHECK-SPIRV: %[[#Int8PtrGen:]] = OpTypePointer Generic %[[#Int8]]
  28 ; CHECK-SPIRV: %[[#Int8Ptr:]] = OpTypePointer CrossWorkgroup %[[#Int8]]
  29 ; CHECK-SPIRV: %[[#Block]] = OpVariable %[[#]]
  30
  31 define spir_kernel void @test_enqueue_empty() {
  32 entry:
  33   %tmp = alloca %struct.ndrange_t, align 8
  34   %call = call spir_func %opencl.queue_t* @_Z17get_default_queuev()
  35   call spir_func void @_Z10ndrange_1Dm(%struct.ndrange_t* sret(%struct.ndrange_t*) %tmp, i64 1)
  36   %0 = call i32 @__enqueue_kernel_basic_events(%opencl.queue_t* %call, i32 1, %struct.ndrange_t* %tmp, i32 0, %opencl.clk_event_t* addrspace(4)* null, %opencl.clk_event_t* addrspace(4)* null, i8 addrspace(4)* addrspacecast (i8* bitcast (void (i8 addrspace(4)*)* @__test_enqueue_empty_block_invoke_kernel to i8*) to i8 addrspace(4)*), i8 addrspace(4)* addrspacecast (i8 addrspace(1)* bitcast ({ i32, i32 } addrspace(1)* @__block_literal_global to i8 addrspace(1)*) to i8 addrspace(4)*))
  37   ret void
  38 ; CHECK-SPIRV: %[[#Int8PtrBlock:]] = OpBitcast %[[#Int8Ptr]] %[[#Block]]
  39 ; CHECK-SPIRV: %[[#Int8PtrGenBlock:]] = OpPtrCastToGeneric %[[#Int8PtrGen]] %[[#Int8PtrBlock]]
  40 ; CHECK-SPIRV: %[[#]] = OpEnqueueKernel %[[#]] %[[#]] %[[#]] %[[#]] %[[#]] %[[#]] %[[#]] %[[#Invoke:]] %[[#Int8PtrGenBlock]] %[[#]] %[[#]]
  41 }
  42
  43 declare spir_func %opencl.queue_t* @_Z17get_default_queuev()
  44
  45 declare spir_func void @_Z10ndrange_1Dm(%struct.ndrange_t* sret(%struct.ndrange_t*), i64)
  46
  47 define internal spir_func void @__test_enqueue_empty_block_invoke(i8 addrspace(4)* %.block_descriptor) {
  48 entry:
  49   %.block_descriptor.addr = alloca i8 addrspace(4)*, align 8
  50   store i8 addrspace(4)* %.block_descriptor, i8 addrspace(4)** %.block_descriptor.addr, align 8
  51   %block = bitcast i8 addrspace(4)* %.block_descriptor to <{ i32, i32 }> addrspace(4)*
  52   ret void
  53 }
  54
  55 define internal spir_kernel void @__test_enqueue_empty_block_invoke_kernel(i8 addrspace(4)*) {
  56 entry:
  57   call void @__test_enqueue_empty_block_invoke(i8 addrspace(4)* %0)
  58   ret void
  59 }
  60
  61 declare i32 @__enqueue_kernel_basic_events(%opencl.queue_t*, i32, %struct.ndrange_t*, i32, %opencl.clk_event_t* addrspace(4)*, %opencl.clk_event_t* addrspace(4)*, i8 addrspace(4)*, i8 addrspace(4)*)
  62
  63 ; CHECK-SPIRV:      %[[#Invoke]] = OpFunction %[[#Void]] None %[[#]]
  64 ; CHECK-SPIRV-NEXT: %[[#]] = OpFunctionParameter %[[#Int8PtrGen]]