cjatin
41 lignes
1.2 KiB
C++
41 lignes
1.2 KiB
C++
#include <hip_test_common.hh>
|
|
#include <iostream>
|
|
|
|
template <typename T> __global__ void add(T* a, T* b, T* c, size_t size) {
|
|
size_t i = threadIdx.x;
|
|
if (i < size) c[i] = a[i] + b[i];
|
|
}
|
|
|
|
TEMPLATE_TEST_CASE("ABM_AddKernel_MultiTypeMultiSize", "", int, long, float, long long, double) {
|
|
auto size = GENERATE(as<size_t>{}, 100, 500, 1000);
|
|
TestType *d_a, *d_b, *d_c;
|
|
auto res = hipMalloc(&d_a, sizeof(TestType) * size);
|
|
REQUIRE(res == hipSuccess);
|
|
res = hipMalloc(&d_b, sizeof(TestType) * size);
|
|
REQUIRE(res == hipSuccess);
|
|
res = hipMalloc(&d_c, sizeof(TestType) * size);
|
|
REQUIRE(res == hipSuccess);
|
|
|
|
std::vector<TestType> a, b, c;
|
|
for (size_t i = 0; i < size; i++) {
|
|
a.push_back(i + 1);
|
|
b.push_back(i + 1);
|
|
c.push_back(2 * (i + 1));
|
|
}
|
|
|
|
res = hipMemcpy(d_a, a.data(), sizeof(TestType) * size, hipMemcpyHostToDevice);
|
|
REQUIRE(res == hipSuccess);
|
|
res = hipMemcpy(d_b, b.data(), sizeof(TestType) * size, hipMemcpyHostToDevice);
|
|
REQUIRE(res == hipSuccess);
|
|
|
|
hipLaunchKernelGGL(add<TestType>, 1, size, 0, 0, d_a, d_b, d_c, size);
|
|
|
|
res = hipMemcpy(a.data(), d_c, sizeof(TestType) * size, hipMemcpyDeviceToHost);
|
|
REQUIRE(res == hipSuccess);
|
|
|
|
hipFree(d_a);
|
|
hipFree(d_b);
|
|
hipFree(d_c);
|
|
REQUIRE(a == c);
|
|
}
|