49 lines · plain
1// RUN: mlir-opt -convert-spirv-to-llvm -split-input-file %s | FileCheck %s2 3//===----------------------------------------------------------------------===//4// spirv.ControlBarrierOp5//===----------------------------------------------------------------------===//6 7// CHECK: llvm.func spir_funccc @_Z22__spirv_ControlBarrieriii(i32, i32, i32) attributes {convergent, no_unwind, will_return}8 9// CHECK-LABEL: @control_barrier10spirv.func @control_barrier() "None" {11 // CHECK: [[EXECUTION:%.*]] = llvm.mlir.constant(2 : i32) : i3212 // CHECK: [[MEMORY:%.*]] = llvm.mlir.constant(2 : i32) : i3213 // CHECK: [[SEMANTICS:%.*]] = llvm.mlir.constant(768 : i32) : i3214 // CHECK: llvm.call spir_funccc @_Z22__spirv_ControlBarrieriii([[EXECUTION]], [[MEMORY]], [[SEMANTICS]]) {convergent, no_unwind, will_return} : (i32, i32, i32) -> ()15 spirv.ControlBarrier <Workgroup>, <Workgroup>, <CrossWorkgroupMemory|WorkgroupMemory>16 17 // CHECK: [[EXECUTION:%.*]] = llvm.mlir.constant(2 : i32) : i3218 // CHECK: [[MEMORY:%.*]] = llvm.mlir.constant(2 : i32) : i3219 // CHECK: [[SEMANTICS:%.*]] = llvm.mlir.constant(256 : i32) : i3220 // CHECK: llvm.call spir_funccc @_Z22__spirv_ControlBarrieriii([[EXECUTION]], [[MEMORY]], [[SEMANTICS]]) {convergent, no_unwind, will_return} : (i32, i32, i32) -> ()21 spirv.ControlBarrier <Workgroup>, <Workgroup>, <WorkgroupMemory>22 spirv.Return23}24 25// -----26 27//===----------------------------------------------------------------------===//28// spirv.INTEL.SplitBarrier29//===----------------------------------------------------------------------===//30 31// CHECK-DAG: llvm.func spir_funccc @_Z33__spirv_ControlBarrierArriveINTELiii(i32, i32, i32) attributes {convergent, no_unwind, will_return}32// CHECK-DAG: llvm.func spir_funccc @_Z31__spirv_ControlBarrierWaitINTELiii(i32, i32, i32) attributes {convergent, no_unwind, will_return}33 34// CHECK-LABEL: @split_barrier35spirv.func @split_barrier() "None" {36 // CHECK: [[EXECUTION:%.*]] = llvm.mlir.constant(2 : i32) : i3237 // CHECK: [[MEMORY:%.*]] = llvm.mlir.constant(2 : i32) : i3238 // CHECK: [[SEMANTICS:%.*]] = llvm.mlir.constant(768 : i32) : i3239 // CHECK: llvm.call spir_funccc @_Z33__spirv_ControlBarrierArriveINTELiii([[EXECUTION]], [[MEMORY]], [[SEMANTICS]]) {convergent, no_unwind, will_return} : (i32, i32, i32) -> ()40 spirv.INTEL.ControlBarrierArrive <Workgroup> <Workgroup> <CrossWorkgroupMemory|WorkgroupMemory>41 42 // CHECK: [[EXECUTION:%.*]] = llvm.mlir.constant(2 : i32) : i3243 // CHECK: [[MEMORY:%.*]] = llvm.mlir.constant(2 : i32) : i3244 // CHECK: [[SEMANTICS:%.*]] = llvm.mlir.constant(256 : i32) : i3245 // CHECK: llvm.call spir_funccc @_Z31__spirv_ControlBarrierWaitINTELiii([[EXECUTION]], [[MEMORY]], [[SEMANTICS]]) {convergent, no_unwind, will_return} : (i32, i32, i32) -> ()46 spirv.INTEL.ControlBarrierWait <Workgroup> <Workgroup> <WorkgroupMemory>47 spirv.Return48}49