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
|
/*
* Copyright (C) 2018-2022 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "opencl/test/unit_test/command_queue/command_enqueue_fixture.h"
#include "opencl/test/unit_test/command_queue/enqueue_fixture.h"
#include "opencl/test/unit_test/fixtures/buffer_fixture.h"
#include "opencl/test/unit_test/mocks/mock_context.h"
namespace NEO {
struct EnqueueReadBufferTypeTest : public CommandEnqueueFixture,
public ::testing::Test {
EnqueueReadBufferTypeTest(void)
: srcBuffer(nullptr) {
}
void SetUp() override {
CommandEnqueueFixture::setUp();
BufferDefaults::context = new MockContext;
srcBuffer.reset(BufferHelper<>::create());
nonZeroCopyBuffer.reset(BufferHelper<BufferUseHostPtr<>>::create());
}
void TearDown() override {
srcBuffer.reset(nullptr);
nonZeroCopyBuffer.reset(nullptr);
delete BufferDefaults::context;
CommandEnqueueFixture::tearDown();
}
protected:
template <typename FamilyType>
void enqueueReadBuffer(cl_bool blocking = CL_TRUE) {
auto retVal = EnqueueReadBufferHelper<>::enqueueReadBuffer(
pCmdQ,
srcBuffer.get(),
blocking);
EXPECT_EQ(CL_SUCCESS, retVal);
parseCommands<FamilyType>(*pCmdQ);
}
std::unique_ptr<Buffer> srcBuffer;
std::unique_ptr<Buffer> nonZeroCopyBuffer;
};
} // namespace NEO
|