1 // Test target codegen - host bc file has to be created first.
2 // RUN: %clang_cc1 -verify -fopenmp -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 -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 -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 -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 
7 #include <stdarg.h>
8 
9 // expected-no-diagnostics
10 extern int printf(const char *, ...);
11 extern int vprintf(const char *, va_list);
12 
13 // Check a simple call to printf end-to-end.
14 // CHECK: [[SIMPLE_PRINTF_TY:%[a-zA-Z0-9_]+]] = type { i32, i64, double }
15 int CheckSimple() {
16     // CHECK: define {{.*}}void [[T1:@__omp_offloading_.+CheckSimple.+]]_worker()
17 #pragma omp target
18   {
19     // Entry point.
20     // CHECK: define {{.*}}void [[T1]]()
21     // Alloca in entry block.
22     // CHECK: [[BUF:%[a-zA-Z0-9_]+]] = alloca [[SIMPLE_PRINTF_TY]]
23 
24     // CHECK: {{call|invoke}} void [[T1]]_worker()
25     // CHECK: br label {{%?}}[[EXIT:.+]]
26     //
27     // CHECK-DAG: [[CMTID:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.tid.x()
28     // CHECK-DAG: [[CMNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
29     // CHECK-DAG: [[CMWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
30     // CHECK: [[IS_MASTER:%.+]] = icmp eq i32 [[CMTID]],
31     // CHECK: br i1 [[IS_MASTER]], label {{%?}}[[MASTER:.+]], label {{%?}}[[EXIT]]
32     //
33     // CHECK: [[MASTER]]
34     // CHECK-DAG: [[MNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
35     // CHECK-DAG: [[MWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
36     // CHECK: [[MTMP1:%.+]] = sub i32 [[MNTH]], [[MWS]]
37     // CHECK: call void @__kmpc_kernel_init(i32 [[MTMP1]]
38 
39     // printf in master-only basic block.
40     // CHECK: [[FMT:%[0-9]+]] = load{{.*}}%fmt
41     const char* fmt = "%d %lld %f";
42     // CHECK: [[PTR0:%[0-9]+]] = getelementptr inbounds [[SIMPLE_PRINTF_TY]], [[SIMPLE_PRINTF_TY]]* [[BUF]], i32 0, i32 0
43     // CHECK: store i32 1, i32* [[PTR0]], align 4
44     // CHECK: [[PTR1:%[0-9]+]] = getelementptr inbounds [[SIMPLE_PRINTF_TY]], [[SIMPLE_PRINTF_TY]]* [[BUF]], i32 0, i32 1
45     // CHECK: store i64 2, i64* [[PTR1]], align 8
46     // CHECK: [[PTR2:%[0-9]+]] = getelementptr inbounds [[SIMPLE_PRINTF_TY]], [[SIMPLE_PRINTF_TY]]* [[BUF]], i32 0, i32 2
47 
48     // CHECK: store double 3.0{{[^,]*}}, double* [[PTR2]], align 8
49     // CHECK: [[BUF_CAST:%[0-9]+]] = bitcast [[SIMPLE_PRINTF_TY]]* [[BUF]] to i8*
50     // CHECK: [[RET:%[0-9]+]] = call i32 @vprintf(i8* [[FMT]], i8* [[BUF_CAST]])
51     printf(fmt, 1, 2ll, 3.0);
52   }
53 
54   return 0;
55 }
56 
57 void CheckNoArgs() {
58     // CHECK: define {{.*}}void [[T2:@__omp_offloading_.+CheckNoArgs.+]]_worker()
59 #pragma omp target
60   {
61     // Entry point.
62     // CHECK: define {{.*}}void [[T2]]()
63 
64     // CHECK: {{call|invoke}} void [[T2]]_worker()
65     // CHECK: br label {{%?}}[[EXIT:.+]]
66     //
67     // CHECK-DAG: [[CMTID:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.tid.x()
68     // CHECK-DAG: [[CMNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
69     // CHECK-DAG: [[CMWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
70     // CHECK: [[IS_MASTER:%.+]] = icmp eq i32 [[CMTID]],
71     // CHECK: br i1 [[IS_MASTER]], label {{%?}}[[MASTER:.+]], label {{%?}}[[EXIT]]
72     //
73     // CHECK: [[MASTER]]
74     // CHECK-DAG: [[MNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
75     // CHECK-DAG: [[MWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
76     // CHECK: [[MTMP1:%.+]] = sub i32 [[MNTH]], [[MWS]]
77     // CHECK: call void @__kmpc_kernel_init(i32 [[MTMP1]]
78 
79     // printf in master-only basic block.
80     // CHECK: call i32 @vprintf({{.*}}, i8* null){{$}}
81     printf("hello, world!");
82   }
83 }
84 
85 // Check that printf's alloca happens in the entry block, not inside the if
86 // statement.
87 int foo;
88 void CheckAllocaIsInEntryBlock() {
89     // CHECK: define {{.*}}void [[T3:@__omp_offloading_.+CheckAllocaIsInEntryBlock.+]]_worker()
90 #pragma omp target
91   {
92     // Entry point.
93     // CHECK: define {{.*}}void [[T3]](
94     // Alloca in entry block.
95     // CHECK: alloca %printf_args
96 
97     // CHECK: {{call|invoke}} void [[T3]]_worker()
98     // CHECK: br label {{%?}}[[EXIT:.+]]
99     //
100     // CHECK-DAG: [[CMTID:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.tid.x()
101     // CHECK-DAG: [[CMNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
102     // CHECK-DAG: [[CMWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
103     // CHECK: [[IS_MASTER:%.+]] = icmp eq i32 [[CMTID]],
104     // CHECK: br i1 [[IS_MASTER]], label {{%?}}[[MASTER:.+]], label {{%?}}[[EXIT]]
105     //
106     // CHECK: [[MASTER]]
107     // CHECK-DAG: [[MNTH:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.ntid.x()
108     // CHECK-DAG: [[MWS:%.+]] = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
109     // CHECK: [[MTMP1:%.+]] = sub i32 [[MNTH]], [[MWS]]
110     // CHECK: call void @__kmpc_kernel_init(i32 [[MTMP1]]
111 
112     if (foo) {
113       printf("%d", 42);
114     }
115   }
116 }
117