| // Make sure that __global__ functions are emitted along with correct |
| // annotations and are added to @llvm.used to prevent their elimination. |
| // REQUIRES: nvptx-registered-target |
| // |
| // RUN: %clang_cc1 %s -triple nvptx-unknown-unknown -fcuda-is-device -emit-llvm -o - | FileCheck %s |
| |
| #include "Inputs/cuda.h" |
| |
| // CHECK-LABEL: define void @device_function |
| extern "C" |
| __device__ void device_function() {} |
| |
| // CHECK-LABEL: define void @global_function |
| extern "C" |
| __global__ void global_function() { |
| // CHECK: call void @device_function |
| device_function(); |
| } |
| |
| // Make sure host-instantiated kernels are preserved on device side. |
| template <typename T> __global__ void templated_kernel(T param) {} |
| // CHECK-DAG: define void @_Z16templated_kernelIiEvT_( |
| |
| namespace { |
| __global__ void anonymous_ns_kernel() {} |
| // CHECK-DAG: define void @_ZN12_GLOBAL__N_119anonymous_ns_kernelEv( |
| } |
| |
| void host_function() { |
| templated_kernel<<<0, 0>>>(0); |
| anonymous_ns_kernel<<<0,0>>>(); |
| } |
| |
| // CHECK: !{{[0-9]+}} = !{void ()* @global_function, !"kernel", i32 1} |
| // CHECK: !{{[0-9]+}} = !{void (i32)* @_Z16templated_kernelIiEvT_, !"kernel", i32 1} |
