forked from OSchip/llvm-project
95 lines
2.8 KiB
Plaintext
95 lines
2.8 KiB
Plaintext
|
// REQUIRES: x86-registered-target
|
||
|
|
// REQUIRES: amdgpu-registered-target
|
||
|
|
|
||
|
|
// RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -fcuda-is-device \
|
||
|
|
// RUN: -emit-llvm -o - -x hip %s | FileCheck \
|
||
|
|
// RUN: -check-prefixes=DEV %s
|
||
|
|
|
||
|
|
// RUN: %clang_cc1 -triple x86_64-gnu-linux \
|
||
|
|
// RUN: -emit-llvm -o - -x hip %s | FileCheck \
|
||
|
|
// RUN: -check-prefixes=HOST %s
|
||
|
|
|
||
|
|
#include "Inputs/cuda.h"
|
||
|
|
|
||
|
|
// Test function scope static device variable, which should not be externalized.
|
||
|
|
// DEV-DAG: @_ZZ6kernelPiPPKiE1w = internal addrspace(4) constant i32 1
|
||
|
|
|
||
|
|
// Check a static device variable referenced by host function is externalized.
|
||
|
|
// DEV-DAG: @_ZL1x = addrspace(1) externally_initialized global i32 0
|
||
|
|
// HOST-DAG: @_ZL1x = internal global i32 undef
|
||
|
|
// HOST-DAG: @[[DEVNAMEX:[0-9]+]] = {{.*}}c"_ZL1x\00"
|
||
|
|
|
||
|
|
static __device__ int x;
|
||
|
|
|
||
|
|
// Check a static device variables referenced only by device functions and kernels
|
||
|
|
// is not externalized.
|
||
|
|
// DEV-DAG: @_ZL2x2 = internal addrspace(1) global i32 0
|
||
|
|
static __device__ int x2;
|
||
|
|
|
||
|
|
// Check a static device variable referenced by host device function is externalized.
|
||
|
|
// DEV-DAG: @_ZL2x3 = addrspace(1) externally_initialized global i32 0
|
||
|
|
static __device__ int x3;
|
||
|
|
|
||
|
|
// Check a static device variable referenced in file scope is externalized.
|
||
|
|
// DEV-DAG: @_ZL2x4 = addrspace(1) externally_initialized global i32 0
|
||
|
|
static __device__ int x4;
|
||
|
|
int& x4_ref = x4;
|
||
|
|
|
||
|
|
// Check a static device variable in anonymous namespace.
|
||
|
|
// DEV-DAG: @_ZN12_GLOBAL__N_12x5E = addrspace(1) externally_initialized global i32 0
|
||
|
|
namespace {
|
||
|
|
static __device__ int x5;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check a static constant variable referenced by host is externalized.
|
||
|
|
// DEV-DAG: @_ZL1y = addrspace(4) externally_initialized global i32 0
|
||
|
|
// HOST-DAG: @_ZL1y = internal global i32 undef
|
||
|
|
// HOST-DAG: @[[DEVNAMEY:[0-9]+]] = {{.*}}c"_ZL1y\00"
|
||
|
|
|
||
|
|
static __constant__ int y;
|
||
|
|
|
||
|
|
// Test static host variable, which should not be externalized nor registered.
|
||
|
|
// HOST-DAG: @_ZL1z = internal global i32 0
|
||
|
|
// DEV-NOT: @_ZL1z
|
||
|
|
static int z;
|
||
|
|
|
||
|
|
// Test static device variable in inline function, which should not be
|
||
|
|
// externalized nor registered.
|
||
|
|
// DEV-DAG: @_ZZ6devfunPPKiE1p = linkonce_odr addrspace(4) constant i32 2, comdat
|
||
|
|
|
||
|
|
inline __device__ void devfun(const int ** b) {
|
||
|
|
const static int p = 2;
|
||
|
|
b[0] = &p;
|
||
|
|
b[1] = &x2;
|
||
|
|
}
|
||
|
|
|
||
|
|
__global__ void kernel(int *a, const int **b) {
|
||
|
|
const static int w = 1;
|
||
|
|
a[0] = x;
|
||
|
|
a[1] = y;
|
||
|
|
a[2] = x2;
|
||
|
|
a[3] = x3;
|
||
|
|
a[4] = x4;
|
||
|
|
a[5] = x5;
|
||
|
|
b[0] = &w;
|
||
|
|
devfun(b);
|
||
|
|
}
|
||
|
|
|
||
|
|
__host__ __device__ void hdf(int *a) {
|
||
|
|
a[0] = x3;
|
||
|
|
}
|
||
|
|
|
||
|
|
int* getDeviceSymbol(int *x);
|
||
|
|
|
||
|
|
void foo(int *a) {
|
||
|
|
getDeviceSymbol(&x);
|
||
|
|
getDeviceSymbol(&x5);
|
||
|
|
getDeviceSymbol(&y);
|
||
|
|
z = 123;
|
||
|
|
}
|
||
|
|
|
||
|
|
// HOST: __hipRegisterVar({{.*}}@_ZL1x {{.*}}@[[DEVNAMEX]]
|
||
|
|
// HOST: __hipRegisterVar({{.*}}@_ZL1y {{.*}}@[[DEVNAMEY]]
|
||
|
|
// HOST-NOT: __hipRegisterVar({{.*}}@_ZZ6kernelPiPPKiE1w
|
||
|
|
// HOST-NOT: __hipRegisterVar({{.*}}@_ZZ6devfunPPKiE1p
|