Merge branch 'privatestaging' of https://github.com/AMDComputeLibraries/HIP-privatestaging into privatestaging

Šī revīzija ir iekļauta:
Ben Sander
2016-03-29 17:12:36 -05:00
revīzija ce4052f09c
3 mainīti faili ar 42 papildinājumiem un 0 dzēšanām
+5
Parādīt failu
@@ -0,0 +1,5 @@
#include"hipDeviceHeader.h"
__host__ __device__ int add(int &a, int &b){
return a+b;
}
+7
Parādīt failu
@@ -0,0 +1,7 @@
#ifndef HIPDEVICEHEADER_H
#define HIPDEVICEHEADER_H
#include"test_common.h"
__host__ __device__ int add(int a, int b);
#endif
+30
Parādīt failu
@@ -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);
}