1 // Test target codegen - host bc file has to be created first.
2 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -fopenmp-cuda-mode -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 -fopenmp-cuda-mode -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 -fopenmp-cuda-mode -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 -fopenmp-cuda-mode -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 -fopenmp-cuda-mode -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 // Check that the execution mode of all 2 target regions on the gpu is set to SPMD Mode.
12 // CHECK-DAG: {{@__omp_offloading_.+l24}}_exec_mode = weak constant i8 0
13 // CHECK-DAG: {{@__omp_offloading_.+l29}}_exec_mode = weak constant i8 0
14 // CHECK-DAG: {{@__omp_offloading_.+l34}}_exec_mode = weak constant i8 0
15 
16 #define N 1000
17 
18 template<typename tx>
19 tx ftemplate(int n) {
20   tx a[N];
21   short aa[N];
22   tx b[10];
23 
24   #pragma omp target simd
25   for(int i = 0; i < n; i++) {
26     a[i] = 1;
27   }
28 
29   #pragma omp target simd
30   for(int i = 0; i < n; i++) {
31     aa[i] += 1;
32   }
33 
34   #pragma omp target simd
35   for(int i = 0; i < 10; i++) {
36     b[i] += 1;
37   }
38 
39   return a[0];
40 }
41 
42 int bar(int n){
43   int a = 0;
44 
45   a += ftemplate<int>(n);
46 
47   return a;
48 }
49 
50 // CHECK-LABEL: define {{.*}}void {{@__omp_offloading_.+l24}}(
51 // CHECK-DAG: [[THREAD_LIMIT:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
52 // CHECK: call void @__kmpc_spmd_kernel_init(i32 [[THREAD_LIMIT]],
53 // CHECK-NOT: call void @__kmpc_for_static_init
54 // CHECK-NOT: call void @__kmpc_for_static_fini
55 // CHECK: call void @__kmpc_spmd_kernel_deinit()
56 // CHECK: ret void
57 
58 // CHECK-LABEL: define {{.*}}void {{@__omp_offloading_.+l29}}(
59 // CHECK-DAG: [[THREAD_LIMIT:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
60 // CHECK: call void @__kmpc_spmd_kernel_init(i32 [[THREAD_LIMIT]],
61 // CHECK-NOT: call void @__kmpc_for_static_init
62 // CHECK-NOT: call void @__kmpc_for_static_fini
63 // CHECK: call void @__kmpc_spmd_kernel_deinit()
64 // CHECK: ret void
65 
66 // CHECK-LABEL: define {{.*}}void {{@__omp_offloading_.+l34}}(
67 // CHECK-DAG: [[THREAD_LIMIT:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
68 // CHECK: call void @__kmpc_spmd_kernel_init(i32 [[THREAD_LIMIT]],
69 // CHECK-NOT: call void @__kmpc_for_static_init
70 // CHECK-NOT: call void @__kmpc_for_static_fini
71 // CHECK: call void @__kmpc_spmd_kernel_deinit()
72 // CHECK: ret void
73 
74 #endif
75