Maneesh Gupta
61 строка
1.3 KiB
C++
61 строка
1.3 KiB
C++
#include <iostream>
|
|
#include "hip/hip_runtime.h"
|
|
#include "hip/hip_runtime_api.h"
|
|
#include "../test_common.h"
|
|
|
|
|
|
#define LEN 1024
|
|
#define SIZE LEN << 2
|
|
|
|
/* HIT_START
|
|
* BUILD: %t %s ../test_common.cpp
|
|
* TEST: %t
|
|
* HIT_END
|
|
*/
|
|
|
|
|
|
__global__ void cpy(uint32_t* Out, uint32_t* In) {
|
|
int tx = threadIdx.x;
|
|
memcpy(Out + tx, In + tx, sizeof(uint32_t));
|
|
}
|
|
|
|
__global__ void set(uint32_t* ptr, uint8_t val, size_t size) {
|
|
int tx = threadIdx.x;
|
|
memset(ptr + tx, val, sizeof(uint32_t));
|
|
}
|
|
|
|
int main() {
|
|
uint32_t *A, *Ad, *B, *Bd;
|
|
uint32_t* Val;
|
|
A = new uint32_t[LEN];
|
|
B = new uint32_t[LEN];
|
|
Val = new uint32_t;
|
|
*Val = 0;
|
|
for (int i = 0; i < LEN; i++) {
|
|
A[i] = i;
|
|
B[i] = 0;
|
|
}
|
|
hipMalloc((void**)&Ad, SIZE);
|
|
hipMalloc((void**)&Bd, SIZE);
|
|
hipMemcpy(Ad, A, SIZE, hipMemcpyHostToDevice);
|
|
|
|
hipLaunchKernelGGL(cpy, dim3(1), dim3(LEN), 0, 0, Bd, Ad);
|
|
|
|
hipMemcpy(B, Bd, SIZE, hipMemcpyDeviceToHost);
|
|
for (int i = LEN - 16; i < LEN; i++) {
|
|
if (A[i] != B[i]) {
|
|
return 0;
|
|
}
|
|
}
|
|
hipLaunchKernelGGL(set, dim3(1), dim3(LEN), 0, 0, Bd, 0x1, LEN);
|
|
|
|
hipMemcpy(B, Bd, SIZE, hipMemcpyDeviceToHost);
|
|
for (int i = LEN - 16; i < LEN; i++) {
|
|
if (0x01010101 != B[i]) {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
passed();
|
|
}
|