36 lines
1.1 KiB
Plaintext
36 lines
1.1 KiB
Plaintext
|
// 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}
|