mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-31 12:11:31 +08:00
8f0bdf95ed
Change-Id: I7de5e0124ea54131bb7ea066fdf94f3d03960463 Signed-off-by: Maciej Dziuban <maciej.dziuban@intel.com> Related-To: NEO-3580
47 lines
1.6 KiB
C++
47 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2019-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "core/unit_tests/utilities/base_object_utils.h"
|
|
#include "runtime/helpers/surface_formats.h"
|
|
#include "test.h"
|
|
#include "unit_tests/mocks/mock_context.h"
|
|
#include "unit_tests/mocks/mock_device.h"
|
|
#include "unit_tests/mocks/mock_memory_manager.h"
|
|
|
|
using namespace NEO;
|
|
|
|
class ImageCompressionTests : public ::testing::Test {
|
|
public:
|
|
class MyMemoryManager : public MockMemoryManager {
|
|
public:
|
|
using MockMemoryManager::MockMemoryManager;
|
|
GraphicsAllocation *allocateGraphicsMemoryForImage(const AllocationData &allocationData) override {
|
|
mockMethodCalled = true;
|
|
capturedImgInfo = *allocationData.imgInfo;
|
|
return OsAgnosticMemoryManager::allocateGraphicsMemoryForImage(allocationData);
|
|
}
|
|
ImageInfo capturedImgInfo = {};
|
|
bool mockMethodCalled = false;
|
|
};
|
|
|
|
void SetUp() override {
|
|
mockDevice = std::make_unique<MockClDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(*platformDevices));
|
|
myMemoryManager = new MyMemoryManager(*mockDevice->getExecutionEnvironment());
|
|
mockDevice->injectMemoryManager(myMemoryManager);
|
|
mockContext = make_releaseable<MockContext>(mockDevice.get());
|
|
}
|
|
|
|
std::unique_ptr<MockClDevice> mockDevice;
|
|
ReleaseableObjectPtr<MockContext> mockContext;
|
|
MyMemoryManager *myMemoryManager = nullptr;
|
|
|
|
cl_image_desc imageDesc = {};
|
|
cl_image_format imageFormat{CL_R, CL_UNSIGNED_INT8};
|
|
cl_mem_flags flags = CL_MEM_READ_WRITE;
|
|
cl_int retVal = CL_SUCCESS;
|
|
};
|