89 lines
2.8 KiB
C++
89 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "core/memory_manager/graphics_allocation.h"
|
|
#include "runtime/program/kernel_info.h"
|
|
#include "unit_tests/mocks/mock_block_kernel_manager.h"
|
|
#include "unit_tests/mocks/mock_graphics_allocation.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace NEO;
|
|
|
|
TEST(BlockKernelManagerTest, pushPrivateSurfaceResizesArray) {
|
|
MockGraphicsAllocation allocation(0, 0);
|
|
KernelInfo *blockInfo = new KernelInfo;
|
|
MockBlockKernelManager blockManager;
|
|
|
|
blockManager.addBlockKernelInfo(blockInfo);
|
|
|
|
EXPECT_EQ(0u, blockManager.blockPrivateSurfaceArray.size());
|
|
|
|
blockManager.pushPrivateSurface(&allocation, 0);
|
|
|
|
EXPECT_EQ(1u, blockManager.blockPrivateSurfaceArray.size());
|
|
}
|
|
|
|
TEST(BlockKernelManagerTest, pushPrivateSurfacePlacesAllocationInCorrectPosition) {
|
|
MockGraphicsAllocation allocation1(0, 0);
|
|
MockGraphicsAllocation allocation2(0, 0);
|
|
KernelInfo *blockInfo = new KernelInfo;
|
|
KernelInfo *blockInfo2 = new KernelInfo;
|
|
MockBlockKernelManager blockManager;
|
|
|
|
blockManager.addBlockKernelInfo(blockInfo);
|
|
blockManager.addBlockKernelInfo(blockInfo2);
|
|
|
|
blockManager.pushPrivateSurface(&allocation1, 0);
|
|
blockManager.pushPrivateSurface(&allocation2, 1);
|
|
|
|
EXPECT_EQ(2u, blockManager.blockPrivateSurfaceArray.size());
|
|
|
|
EXPECT_EQ(&allocation1, blockManager.blockPrivateSurfaceArray[0]);
|
|
EXPECT_EQ(&allocation2, blockManager.blockPrivateSurfaceArray[1]);
|
|
}
|
|
|
|
TEST(BlockKernelManagerTest, pushPrivateSurfaceSetsPrivateSurfaceArrayToNullptrOnFirstCall) {
|
|
MockGraphicsAllocation allocation(0, 0);
|
|
KernelInfo *blockInfo = new KernelInfo;
|
|
KernelInfo *blockInfo2 = new KernelInfo;
|
|
KernelInfo *blockInfo3 = new KernelInfo;
|
|
MockBlockKernelManager blockManager;
|
|
|
|
blockManager.addBlockKernelInfo(blockInfo);
|
|
blockManager.addBlockKernelInfo(blockInfo2);
|
|
blockManager.addBlockKernelInfo(blockInfo3);
|
|
|
|
blockManager.pushPrivateSurface(&allocation, 1);
|
|
|
|
EXPECT_EQ(3u, blockManager.blockPrivateSurfaceArray.size());
|
|
|
|
EXPECT_EQ(nullptr, blockManager.blockPrivateSurfaceArray[0]);
|
|
EXPECT_EQ(nullptr, blockManager.blockPrivateSurfaceArray[2]);
|
|
}
|
|
|
|
TEST(BlockKernelManagerTest, getPrivateSurface) {
|
|
MockGraphicsAllocation allocation(0, 0);
|
|
KernelInfo *blockInfo = new KernelInfo;
|
|
|
|
MockBlockKernelManager blockManager;
|
|
|
|
blockManager.addBlockKernelInfo(blockInfo);
|
|
|
|
blockManager.pushPrivateSurface(&allocation, 0);
|
|
|
|
EXPECT_EQ(1u, blockManager.blockPrivateSurfaceArray.size());
|
|
|
|
EXPECT_EQ(&allocation, blockManager.getPrivateSurface(0));
|
|
}
|
|
|
|
TEST(BlockKernelManagerTest, getPrivateSurfaceWithOutOfBoundOrdinalRetunrsNullptr) {
|
|
MockBlockKernelManager blockManager;
|
|
EXPECT_EQ(nullptr, blockManager.getPrivateSurface(0));
|
|
EXPECT_EQ(nullptr, blockManager.getPrivateSurface(10));
|
|
}
|