/* * Copyright (C) 2017-2018 Intel Corporation * * SPDX-License-Identifier: MIT * */ #pragma once #include "gtest/gtest.h" #include "runtime/helpers/ptr_math.h" #include "runtime/helpers/aligned_memory.h" #include "unit_tests/command_queue/command_enqueue_fixture.h" #include "unit_tests/fixtures/buffer_fixture.h" #include "unit_tests/gen_common/gen_cmd_parse.h" #include "unit_tests/mocks/mock_context.h" namespace OCLRT { struct EnqueueReadBufferRectTest : public CommandEnqueueFixture, public ::testing::Test { EnqueueReadBufferRectTest(void) : buffer(nullptr), hostPtr(nullptr) { } void SetUp() override { CommandEnqueueFixture::SetUp(); context.reset(new MockContext(&pCmdQ->getDevice())); BufferDefaults::context = context.get(); //For 3D hostPtr = ::alignedMalloc(slicePitch * rowPitch, 4096); auto retVal = CL_INVALID_VALUE; buffer.reset(Buffer::create( context.get(), CL_MEM_READ_WRITE, slicePitch * rowPitch, nullptr, retVal)); ASSERT_NE(nullptr, buffer.get()); nonZeroCopyBuffer.reset(BufferHelper>::create()); } void TearDown() override { nonZeroCopyBuffer.reset(nullptr); buffer.reset(nullptr); ::alignedFree(hostPtr); context.reset(); CommandEnqueueFixture::TearDown(); } protected: template void enqueueReadBufferRect2D(cl_bool blocking = CL_FALSE) { typedef typename FamilyType::WALKER_TYPE GPGPU_WALKER; typedef typename FamilyType::PIPELINE_SELECT PIPELINE_SELECT; size_t bufferOrigin[] = {0, 0, 0}; size_t hostOrigin[] = {0, 0, 0}; size_t region[] = {50, 50, 1}; auto retVal = pCmdQ->enqueueReadBufferRect( buffer.get(), blocking, //non-blocking bufferOrigin, hostOrigin, region, rowPitch, slicePitch, rowPitch, slicePitch, hostPtr, 0, nullptr, nullptr); EXPECT_EQ(CL_SUCCESS, retVal); parseCommands(*pCmdQ); } std::unique_ptr context; std::unique_ptr buffer; std::unique_ptr nonZeroCopyBuffer; void *hostPtr; static const size_t rowPitch = 100; static const size_t slicePitch = 100 * 100; }; } // namespace OCLRT