flang/test/Integration/OpenMP/target-use-device-nested.f90

!===----------------------------------------------------------------------===!
! This directory can be used to add Integration tests involving multiple
! stages of the compiler (for eg. from Fortran to LLVM IR). It should not
! contain executable tests. We should only add tests here sparingly and only
! if there is no other way to test. Repeat this message in each test that is
! added to this directory and sub-directories.
!===----------------------------------------------------------------------===!

! This tests check that target code nested inside a target data region which
! has only use_device_ptr mapping corectly generates code on the device pass.

!REQUIRES: amdgpu-registered-target
!RUN: %flang_fc1 -triple amdgcn-amd-amdhsa -emit-llvm -fopenmp -fopenmp-version=50 -fopenmp-is-target-device %s -o - | FileCheck %s

program main
  use iso_c_binding
  implicit none
  type(c_ptr) :: a
  !$omp target data use_device_ptr(a)
    !$omp target map(tofrom: a)
      call foo(a)
    !$omp end target
  !$omp end target data
end program

! CHECK:         define weak_odr protected amdgpu_kernel void @__omp_offloading{{.*}}main_
! CHECK-NEXT:       entry:
! CHECK-NEXT:         %[[VAL_3:.*]] = alloca ptr, align 8, addrspace(5)
! CHECK-NEXT:         %[[ASCAST:.*]] = addrspacecast ptr addrspace(5) %[[VAL_3]] to ptr
! CHECK-NEXT:         store ptr %[[VAL_4:.*]], ptr %[[ASCAST]], align 8
! CHECK-NEXT:         %[[VAL_5:.*]] = call i32 @__kmpc_target_init(ptr addrspacecast (ptr addrspace(1) @__omp_offloading_{{.*}}_kernel_environment to ptr), ptr %[[VAL_6:.*]])
! CHECK-NEXT:         %[[VAL_7:.*]] = icmp eq i32 %[[VAL_5]], -1
! CHECK-NEXT:         br i1 %[[VAL_7]], label %[[VAL_8:.*]], label %[[VAL_9:.*]]
! CHECK:            user_code.entry:                                  ; preds = %[[VAL_10:.*]]
! CHECK-NEXT:         %[[VAL_11:.*]] = load ptr, ptr %[[ASCAST]], align 8
! CHECK-NEXT:         br label %[[AFTER_ALLOC:.*]]

! CHECK:            [[AFTER_ALLOC]]:
! CHECK-NEXT:         br label %[[VAL_12:.*]]

! CHECK:            [[VAL_12]]:
! CHECK-NEXT:         br label %[[TARGET_REG_ENTRY:.*]]

! CHECK:            [[TARGET_REG_ENTRY]]:                                       ; preds = %[[VAL_12]]
! CHECK-NEXT:         call void @{{.*}}foo{{.*}}(ptr %[[VAL_11]])
! CHECK-NEXT:         br label