mlir/test/Target/LLVMIR/target-ext-type.mlir

   1 // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s
   2
   3 // CHECK: @global = global target("spirv.DeviceEvent") zeroinitializer
   4 llvm.mlir.global external @global() {addr_space = 0 : i32} : !llvm.target<"spirv.DeviceEvent"> {
   5   %0 = llvm.mlir.zero : !llvm.target<"spirv.DeviceEvent">
   6   llvm.return %0 : !llvm.target<"spirv.DeviceEvent">
   7 }
   8
   9 // CHECK-LABEL: define target("spirv.Event") @func2() {
  10 // CHECK-NEXT:    %1 = alloca target("spirv.Event"), align 8
  11 // CHECK-NEXT:    %2 = load target("spirv.Event"), ptr %1, align 8
  12 // CHECK-NEXT:    ret target("spirv.Event") poison
  13 llvm.func @func2() -> !llvm.target<"spirv.Event"> {
  14   %0 = llvm.mlir.constant(1 : i32) : i32
  15   %1 = llvm.mlir.poison : !llvm.target<"spirv.Event">
  16   %2 = llvm.alloca %0 x !llvm.target<"spirv.Event"> {alignment = 8 : i64} : (i32) -> !llvm.ptr
  17   %3 = llvm.load %2 {alignment = 8 : i64} : !llvm.ptr -> !llvm.target<"spirv.Event">
  18   llvm.return %1 : !llvm.target<"spirv.Event">
  19 }
  20
  21 // CHECK-LABEL: define void @func3() {
  22 // CHECK-NEXT:    %1 = freeze target("spirv.DeviceEvent") zeroinitializer
  23 // CHECK-NEXT:    ret void
  24 llvm.func @func3() {
  25   %0 = llvm.mlir.zero : !llvm.target<"spirv.DeviceEvent">
  26   %1 = llvm.freeze %0 : !llvm.target<"spirv.DeviceEvent">
  27   llvm.return
  28 }