opencl_test.cpp
1.99 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
//===--- opencl_test.cpp - Tests for OpenCL and the Acxxel API ------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "acxxel.h"
#include "gtest/gtest.h"
#include <array>
#include <cstring>
namespace {
static const char *SaxpyKernelSource = R"(
__kernel void saxpyKernel(float A, __global float *X, __global float *Y, int N) {
int I = get_global_id(0);
if (I < N)
X[I] = A * X[I] + Y[I];
}
)";
TEST(OpenCL, Saxpy) {
constexpr size_t Length = 3;
float A = 2.f;
std::array<float, Length> X = {{0.f, 1.f, 2.f}};
std::array<float, Length> Y = {{3.f, 4.f, 5.f}};
std::array<float, Length> Expected = {{3.f, 6.f, 9.f}};
acxxel::Platform *OpenCL = acxxel::getOpenCLPlatform().getValue();
acxxel::Stream Stream = OpenCL->createStream().takeValue();
auto DeviceX = OpenCL->mallocD<float>(Length).takeValue();
auto DeviceY = OpenCL->mallocD<float>(Length).takeValue();
Stream.syncCopyHToD(X, DeviceX);
Stream.syncCopyHToD(Y, DeviceY);
acxxel::Program Program =
OpenCL
->createProgramFromSource(acxxel::Span<const char>(
SaxpyKernelSource, std::strlen(SaxpyKernelSource)))
.takeValue();
acxxel::Kernel Kernel = Program.createKernel("saxpyKernel").takeValue();
float *RawX = static_cast<float *>(DeviceX);
float *RawY = static_cast<float *>(DeviceY);
int IntLength = Length;
void *Arguments[] = {&A, &RawX, &RawY, &IntLength};
size_t ArgumentSizes[] = {sizeof(float), sizeof(float *), sizeof(float *),
sizeof(int)};
EXPECT_FALSE(
Stream.asyncKernelLaunch(Kernel, Length, Arguments, ArgumentSizes)
.takeStatus()
.isError());
Stream.syncCopyDToH(DeviceX, X);
EXPECT_FALSE(Stream.sync().isError());
EXPECT_EQ(X, Expected);
}
} // namespace