Maneesh Gupta
23 خطوط
549 B
C++
23 خطوط
549 B
C++
#include "hip/hip_runtime.h"
|
|
#include<stdio.h>
|
|
|
|
#define ITER 1<<20
|
|
#define SIZE 1024*1024*sizeof(int)
|
|
|
|
__global__ void Iter(hipLaunchParm lp, int *Ad){
|
|
int tx = hipThreadIdx_x + hipBlockIdx_x * hipBlockDim_x;
|
|
if(tx == 0){
|
|
for(int i=0;i<ITER;i++){
|
|
Ad[tx] += 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(){
|
|
int A=0, *Ad;
|
|
hipMalloc((void**)&Ad, SIZE);
|
|
hipMemcpy(Ad, &A, SIZE, hipMemcpyHostToDevice);
|
|
hipLaunchKernel(HIP_KERNEL_NAME(Iter), dim3(1), dim3(1), 0, 0, Ad);
|
|
hipMemcpy(&A, Ad, SIZE, hipMemcpyDeviceToHost);
|
|
}
|