xref: /llvm-project/clang/test/Interpreter/CUDA/host-and-device.cu (revision ddeab07ca63235f8d952e1171b56fdb0f2d761c9)
1 // Checks that a function is available in both __host__ and __device__
2 // RUN: cat %s | clang-repl --cuda | FileCheck %s
3 
4 extern "C" int printf(const char*, ...);
5 
sum(int a,int b)6 __host__ __device__ inline int sum(int a, int b){ return a + b; }
kernel(int * output)7 __global__ void kernel(int * output){ *output = sum(40,2); }
8 
9 printf("Host sum: %d\n", sum(41,1));
10 // CHECK: Host sum: 42
11 
12 int var = 0;
13 int * deviceVar;
14 printf("cudaMalloc: %d\n", cudaMalloc((void **) &deviceVar, sizeof(int)));
15 // CHECK-NEXT: cudaMalloc: 0
16 
17 kernel<<<1,1>>>(deviceVar);
18 printf("CUDA Error: %d\n", cudaGetLastError());
19 // CHECK-NEXT: CUDA Error: 0
20 
21 printf("cudaMemcpy: %d\n", cudaMemcpy(&var, deviceVar, sizeof(int), cudaMemcpyDeviceToHost));
22 // CHECK-NEXT: cudaMemcpy: 0
23 
24 printf("var: %d\n", var);
25 // CHECK-NEXT: var: 42
26 
27 %quit
28