1 // RUN: %clang_cc1 -no-opaque-pointers %s --std=c++11 -triple nvptx-unknown-unknown -fcuda-is-device \
2 // RUN: -emit-llvm -o - | FileCheck %s --check-prefixes=COMMON,DEVICE
3 // RUN: %clang_cc1 -no-opaque-pointers %s --std=c++11 -triple nvptx-unknown-unknown \
4 // RUN: -emit-llvm -o - | FileCheck %s --check-prefixes=COMMON,HOST
5 // RUN: %clang_cc1 -no-opaque-pointers %s --std=c++17 -triple nvptx-unknown-unknown -fcuda-is-device \
6 // RUN: -emit-llvm -o - | FileCheck %s --check-prefixes=COMMON,DEVICE
7 // RUN: %clang_cc1 -no-opaque-pointers %s --std=c++17 -triple nvptx-unknown-unknown \
8 // RUN: -emit-llvm -o - | FileCheck %s --check-prefixes=COMMON,HOST
9
10 #include "Inputs/cuda.h"
11 extern "C" __host__ void host_fn();
12 extern "C" __device__ void dev_fn();
13 extern "C" __host__ __device__ void hd_fn();
14
15 // Destructors are handled a bit differently, compared to regular functions.
16 // Make sure we do trigger kernel generation on the GPU side even if it's only
17 // referenced by the destructor.
f(T)18 template<typename T> __global__ void f(T) {}
19 template<typename T> struct A {
~AA20 ~A() { f<<<1, 1>>>(T()); }
21 };
22
23 // HOST-LABEL: @a
24 A<int> a;
25 // HOST-LABEL: define linkonce_odr void @_ZN1AIiED1Ev
26 // search further down for the deice-side checks for @_Z1fIiEvT_
27
28 struct H1D1 {
operator deleteH1D129 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1D130 __device__ void operator delete(void *) { dev_fn(); };
31 };
32
33 struct H1D2 {
operator deleteH1D234 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1D235 __device__ void operator delete(void *, __SIZE_TYPE__) { dev_fn(); };
36 };
37
38 struct H2D1 {
operator deleteH2D139 __host__ void operator delete(void *, __SIZE_TYPE__) { host_fn(); };
operator deleteH2D140 __device__ void operator delete(void *) { dev_fn(); };
41 };
42
43 struct H2D2 {
operator deleteH2D244 __host__ void operator delete(void *, __SIZE_TYPE__) { host_fn(); };
operator deleteH2D245 __device__ void operator delete(void *, __SIZE_TYPE__) { dev_fn(); };
46 };
47
48 struct H1D1D2 {
operator deleteH1D1D249 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1D1D250 __device__ void operator delete(void *) { dev_fn(); };
operator deleteH1D1D251 __device__ void operator delete(void *, __SIZE_TYPE__) { dev_fn(); };
52 };
53
54 struct H1H2D1 {
operator deleteH1H2D155 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1H2D156 __host__ void operator delete(void *, __SIZE_TYPE__) { host_fn(); };
operator deleteH1H2D157 __device__ void operator delete(void *) { dev_fn(); };
58 };
59
60 struct H1H2D2 {
operator deleteH1H2D261 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1H2D262 __host__ void operator delete(void *, __SIZE_TYPE__) { host_fn(); };
operator deleteH1H2D263 __device__ void operator delete(void *, __SIZE_TYPE__) { dev_fn(); };
64 };
65
66 struct H1H2D1D2 {
operator deleteH1H2D1D267 __host__ void operator delete(void *) { host_fn(); };
operator deleteH1H2D1D268 __host__ void operator delete(void *, __SIZE_TYPE__) { host_fn(); };
operator deleteH1H2D1D269 __device__ void operator delete(void *) { dev_fn(); };
operator deleteH1H2D1D270 __device__ void operator delete(void *, __SIZE_TYPE__) { dev_fn(); };
71 };
72
73
74 template <typename T>
test_hd(void * p)75 __host__ __device__ void test_hd(void *p) {
76 T *t = (T *)p;
77 delete t;
78 }
79
80 // Make sure we call the right variant of usual deallocator.
tests_hd(void * t)81 __host__ __device__ void tests_hd(void *t) {
82 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI4H1D1EvPv
83 // COMMON: call void @_ZN4H1D1dlEPv
84 test_hd<H1D1>(t);
85 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI4H1D2EvPv
86 // DEVICE: call void @_ZN4H1D2dlEPvj(i8* noundef {{.*}}, i32 noundef 1)
87 // HOST: call void @_ZN4H1D2dlEPv(i8* noundef {{.*}})
88 test_hd<H1D2>(t);
89 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI4H2D1EvPv
90 // DEVICE: call void @_ZN4H2D1dlEPv(i8* {{.*}})
91 // HOST: call void @_ZN4H2D1dlEPvj(i8* noundef %3, i32 noundef 1)
92 test_hd<H2D1>(t);
93 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI4H2D2EvPv
94 // COMMON: call void @_ZN4H2D2dlEPvj(i8* noundef {{.*}}, i32 noundef 1)
95 test_hd<H2D2>(t);
96 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI6H1D1D2EvPv
97 // COMMON: call void @_ZN6H1D1D2dlEPv(i8* noundef %3)
98 test_hd<H1D1D2>(t);
99 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI6H1H2D1EvPv
100 // COMMON: call void @_ZN6H1H2D1dlEPv(i8* {{.*}})
101 test_hd<H1H2D1>(t);
102 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI6H1H2D2EvPv
103 // DEVICE: call void @_ZN6H1H2D2dlEPvj(i8* noundef {{.*}}, i32 noundef 1)
104 // HOST: call void @_ZN6H1H2D2dlEPv(i8* noundef {{.*}})
105 test_hd<H1H2D2>(t);
106 // COMMON-LABEL: define linkonce_odr void @_Z7test_hdI8H1H2D1D2EvPv
107 // COMMON: call void @_ZN8H1H2D1D2dlEPv(i8* {{.*}})
108 test_hd<H1H2D1D2>(t);
109 }
110
111 // Make sure that we've generated the kernel used by A::~A.
112 // DEVICE-LABEL: define void @_Z1fIiEvT_
113
114 // Make sure we've picked deallocator for the correct side of compilation.
115
116 // COMMON-LABEL: define linkonce_odr void @_ZN4H1D1dlEPv(i8* noundef %0)
117 // DEVICE: call void @dev_fn()
118 // HOST: call void @host_fn()
119
120 // DEVICE-LABEL: define linkonce_odr void @_ZN4H1D2dlEPvj(i8* noundef %0, i32 noundef %1)
121 // DEVICE: call void @dev_fn()
122 // HOST-LABEL: define linkonce_odr void @_ZN4H1D2dlEPv(i8* noundef %0)
123 // HOST: call void @host_fn()
124
125 // DEVICE-LABEL: define linkonce_odr void @_ZN4H2D1dlEPv(i8* noundef %0)
126 // DEVICE: call void @dev_fn()
127 // HOST-LABEL: define linkonce_odr void @_ZN4H2D1dlEPvj(i8* noundef %0, i32 noundef %1)
128 // HOST: call void @host_fn()
129
130 // COMMON-LABEL: define linkonce_odr void @_ZN4H2D2dlEPvj(i8* noundef %0, i32 noundef %1)
131 // DEVICE: call void @dev_fn()
132 // HOST: call void @host_fn()
133
134 // COMMON-LABEL: define linkonce_odr void @_ZN6H1D1D2dlEPv(i8* noundef %0)
135 // DEVICE: call void @dev_fn()
136 // HOST: call void @host_fn()
137
138 // COMMON-LABEL: define linkonce_odr void @_ZN6H1H2D1dlEPv(i8* noundef %0)
139 // DEVICE: call void @dev_fn()
140 // HOST: call void @host_fn()
141
142 // DEVICE-LABEL: define linkonce_odr void @_ZN6H1H2D2dlEPvj(i8* noundef %0, i32 noundef %1)
143 // DEVICE: call void @dev_fn()
144 // HOST-LABEL: define linkonce_odr void @_ZN6H1H2D2dlEPv(i8* noundef %0)
145 // HOST: call void @host_fn()
146
147 // COMMON-LABEL: define linkonce_odr void @_ZN8H1H2D1D2dlEPv(i8* noundef %0)
148 // DEVICE: call void @dev_fn()
149 // HOST: call void @host_fn()
150
151 // DEVICE: !0 = !{void (i32)* @_Z1fIiEvT_, !"kernel", i32 1}
152