Added tests for running device code from a difference source file

This commit is contained in:
Aditya Atluri
2016-03-29 09:22:31 -05:00
parent 1991fb5c92
commit 2f46ada042
3 changed files with 41 additions and 0 deletions
+5
View File
@@ -0,0 +1,5 @@
#include"hipDeviceHeader.h"
__device__ int add(int &a, int &b){
return a+b;
}
+6
View File
@@ -0,0 +1,6 @@
#ifndef HIPDEVICEHEADER_H
#define HIPDEVICEHEADER_H
#include"test_common.h"
__device__ int add(int a, int b);
#endif
+30
View File
@@ -0,0 +1,30 @@
#include"hipDeviceHeader.h"
#define LEN 1024*1024
#define SIZE LEN * sizeof(int)
__global__ void deviceAdd(hipLaunchParm lp, int *Ad, int *Bd, int *Cd){
int tx = hipThreadIdx_x + hipBlockIdx_x * hipBlockDim_x;
Cd[tx] = add(Ad[tx], Bd[tx]);
}
int main(){
int *A, *B, *C, *Ad, *Bd, *Cd;
A = new int[LEN];
B = new int[LEN];
C = new int[LEN];
for(int i=0;i<LEN;i++){
A[i] = 1;
B[i] = 1;
}
HIPCHECK(hipMalloc((void**)&Ad, SIZE));
HIPCHECK(hipMalloc((void**)&Bd, SIZE));
HIPCHECK(hipMalloc((void**)&Cd, SIZE));
HIPCHECK(hipMemcpy(Ad, A, SIZE, hipMemcpyHostToDevice));
HIPCHECK(hipMemcpy(Bd, B, SIZE, hipMemcpyHostToDevice));
hipLaunchKernel(HIP_KERNEL_NAME(deviceAdd), dim3(LEN/1024), dim3(1024), 0, 0, Ad, Bd, Cd);
}