Aditya Atluri
45 行
913 B
C++
45 行
913 B
C++
#include<iostream>
|
|
#include<hip/hip_runtime.h>
|
|
#include<hip/hip_runtime_api.h>
|
|
|
|
#define LEN 1024
|
|
#define SIZE LEN<<2
|
|
|
|
int main(){
|
|
int *A, *B;
|
|
hipDeviceptr_t Ad, Bd;
|
|
A = new int[LEN];
|
|
B = new int[LEN];
|
|
|
|
for(int i=0;i<LEN;i++){
|
|
A[i] = i;
|
|
}
|
|
|
|
hipMalloc((void**)&Ad, SIZE);
|
|
hipMalloc((void**)&Bd, SIZE);
|
|
|
|
hipMemcpyHtoD(Ad, A, SIZE);
|
|
hipMemcpyDtoD(Bd, Ad, SIZE);
|
|
hipMemcpyDtoH(B, Bd, SIZE);
|
|
|
|
for(int i=0;i<16;i++){
|
|
std::cout<<A[i]<<" "<<B[i]<<std::endl;
|
|
}
|
|
|
|
int *Ah, *Bh;
|
|
hipHostMalloc(&Ah, SIZE, 0);
|
|
hipHostMalloc(&Bh, SIZE, 0);
|
|
memcpy(Ah, A, SIZE);
|
|
hipStream_t stream;
|
|
hipStreamCreate(&stream);
|
|
|
|
hipMemcpyHtoDAsync(Ad, Ah, SIZE, stream);
|
|
hipStreamSynchronize(stream);
|
|
hipMemcpyDtoDAsync(Bd, Ad, SIZE, stream);
|
|
hipStreamSynchronize(stream);
|
|
hipMemcpyDtoHAsync(Bh, Bd, SIZE, stream);
|
|
hipStreamSynchronize(stream);
|
|
|
|
std::cout<<Ah[10]<<" "<<Bh[10]<<std::endl;
|
|
}
|