// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s

// The aim of the test is to check the LLVM IR codegen for the device
// for omp teams construct

module attributes {omp.is_target_device = true} {
  llvm.func @foo(i32)
  llvm.func @omp_target_teams_shared_simple(%arg0 : i32) attributes {
    omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>
  } {
    omp.teams {
      llvm.call @foo(%arg0) : (i32) -> ()
      omp.terminator
    }
  llvm.return
  }
}

// CHECK-LABEL: @omp_target_teams_shared_simple
// CHECK-SAME: (i32 [[ARG0:%.+]])
// CHECK: call void @[[OUTLINED_FN:.*]](
// CHECK-NOT: call {{.+}} @__kmpc_fork_teams
// CHECK: ret void

//CHECK: define internal void @[[OUTLINED_FN]](
//CHECK: call void @foo(i32 %[[FOO_ARG:.*]])
//CHECK: ret void