test/OpenMP/nvptx_target_teams_distribute_parallel_for_generic_mode_codegen.cpp

   1 // Test target codegen - host bc file has to be created first.
   2 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm-bc %s -o %t-ppc-host.bc
   3 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -x c++ -triple nvptx64-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-64
   4 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -x c++ -triple i386-unknown-unknown -fopenmp-targets=nvptx-nvidia-cuda -emit-llvm-bc %s -o %t-x86-host.bc
   5 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -x c++ -triple nvptx-unknown-unknown -fopenmp-targets=nvptx-nvidia-cuda -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-x86-host.bc -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-32
   6 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -fexceptions -fcxx-exceptions -x c++ -triple nvptx-unknown-unknown -fopenmp-targets=nvptx-nvidia-cuda -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-x86-host.bc -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-32
   7 // expected-no-diagnostics
   8 #ifndef HEADER
   9 #define HEADER
  10
  11 int a;
  12
  13 int foo(int *a);
  14
  15 int main(int argc, char **argv) {
  16 #pragma omp target teams distribute parallel for map(tofrom:a) if(target:argc) schedule(static, a)
  17   for (int i= 0; i < argc; ++i)
  18     a = foo(&i) + foo(&a) + foo(&argc);
  19   return 0;
  20 }
  21
  22 // CHECK: @__omp_offloading_{{.*}}_main_l16_exec_mode = weak constant i8 0
  23
  24 // CHECK: define weak void @__omp_offloading_{{.*}}_main_l16(i{{64|32}} %{{[^,].*}}, i32* dereferenceable{{[^,]*}}, i{{64|32}} %{{[^,)]*}})
  25 // CHECK: call void @__kmpc_spmd_kernel_init(
  26 // CHECK: [[TID:%.+]] = call i32 @__kmpc_global_thread_num(%struct.ident_t* @
  27 // CHECK: call void @__kmpc_for_static_init_4(
  28
  29 // CHECK: call void [[PARALLEL:@.+]](i32* %{{.*}}, i32* %{{.+}}, i{{64|32}} %{{.+}}, i{{64|32}} %{{.*}}, i{{64|32}} %{{.*}}, i32* %{{.*}})
  30 // CHECK: br label %
  31
  32
  33 // CHECK: call void @__kmpc_for_static_fini(%struct.ident_t* @
  34
  35 // CHECK: call void @__kmpc_spmd_kernel_deinit_v2(i16 0)
  36
  37 // CHECK: define internal void [[PARALLEL]](i32* noalias %{{.+}}, i32* noalias %{{.+}}, i{{64|32}} %{{.+}}, i{{64|32}} %{{.+}}, i{{64|32}} [[ARGC:%.+]], i32* dereferenceable{{.*}})
  38 // CHECK-NOT: call i8* @__kmpc_data_sharing_push_stack(
  39 // CHECK: alloca i{{[0-9]+}},
  40 // CHECK: alloca i{{[0-9]+}},
  41 // CHECK: [[ARGC_ADDR:%.+]] = alloca i{{32|64}},
  42 // CHECK: alloca i{{[0-9]+}},
  43 // CHECK: alloca i{{[0-9]+}},
  44 // CHECK: alloca i{{[0-9]+}},
  45 // CHECK: alloca i{{[0-9]+}},
  46 // CHECK: alloca i{{[0-9]+}},
  47 // CHECK: alloca i{{[0-9]+}},
  48 // CHECK: alloca i{{[0-9]+}},
  49 // CHECK: alloca i{{[0-9]+}},
  50 // CHECK: alloca i{{[0-9]+}},
  51 // CHECK: alloca i{{[0-9]+}},
  52 // CHECK: [[I:%.+]] = alloca i32,
  53 // CHECK-32: store i32 [[ARGC]], i32* [[ARGC_ADDR]],
  54 // CHECK-64: store i{{64|32}} [[ARGC]], i{{64|32}}* [[ARGC_ADDR]],
  55 // CHECK-64: [[ARGC:%.+]] = bitcast i64* [[ARGC_ADDR]] to i32*
  56
  57 // CHECK: call void @__kmpc_for_static_init_4(
  58 // CHECK: call i32 [[FOO:@.+foo.+]](i32* [[I]])
  59 // CHECK: call i32 [[FOO]](i32* %{{.+}})
  60 // CHECK-32: call i32 [[FOO]](i32* [[ARGC_ADDR]])
  61 // CHECK-64: call i32 [[FOO]](i32* [[ARGC]])
  62 // CHECK: call void @__kmpc_for_static_fini(
  63
  64 // CHECK-NOT: call void @__kmpc_data_sharing_pop_stack(
  65
  66 #endif