139 lines
4.5 KiB
C++
139 lines
4.5 KiB
C++
/*
|
|
* Copyright (C) 2018-2024 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/indirect_heap/indirect_heap.h"
|
|
#include "shared/source/memory_manager/memory_manager.h"
|
|
#include "shared/test/common/test_macros/hw_test.h"
|
|
|
|
#include "opencl/source/command_queue/command_queue.h"
|
|
#include "opencl/source/event/event.h"
|
|
#include "opencl/test/unit_test/command_queue/command_queue_fixture.h"
|
|
#include "opencl/test/unit_test/fixtures/cl_device_fixture.h"
|
|
#include "opencl/test/unit_test/mocks/mock_cl_device.h"
|
|
#include "opencl/test/unit_test/mocks/mock_context.h"
|
|
|
|
using namespace NEO;
|
|
|
|
struct OOMSetting {
|
|
bool oomCS;
|
|
bool oomISH;
|
|
};
|
|
|
|
static OOMSetting oomSettings[] = {
|
|
{true, false},
|
|
{false, true},
|
|
{true, true}};
|
|
|
|
struct OOMCommandQueueTest : public ClDeviceFixture,
|
|
public CommandQueueFixture,
|
|
public ::testing::TestWithParam<OOMSetting> {
|
|
|
|
using CommandQueueFixture::setUp;
|
|
|
|
OOMCommandQueueTest() {
|
|
}
|
|
|
|
void SetUp() override {
|
|
ClDeviceFixture::setUp();
|
|
context = new MockContext(pClDevice);
|
|
CommandQueueFixture::setUp(context, pClDevice, 0);
|
|
|
|
const auto &oomSetting = GetParam();
|
|
auto oomSize = 10u;
|
|
if (oomSetting.oomCS) {
|
|
auto &cs = pCmdQ->getCS(oomSize);
|
|
|
|
// CommandStream may be larger than requested so grab what wasnt requested
|
|
cs.getSpace(cs.getAvailableSpace() - oomSize);
|
|
ASSERT_EQ(oomSize, cs.getAvailableSpace());
|
|
}
|
|
|
|
if (oomSetting.oomISH) {
|
|
auto &ish = pCmdQ->getIndirectHeap(IndirectHeap::Type::dynamicState, oomSize);
|
|
|
|
// IndirectHeap may be larger than requested so grab what wasnt requested
|
|
ish.getSpace(ish.getAvailableSpace() - oomSize);
|
|
ASSERT_EQ(oomSize, ish.getAvailableSpace());
|
|
}
|
|
}
|
|
|
|
void TearDown() override {
|
|
CommandQueueFixture::tearDown();
|
|
context->release();
|
|
ClDeviceFixture::tearDown();
|
|
}
|
|
|
|
MockContext *context;
|
|
};
|
|
|
|
HWTEST_P(OOMCommandQueueTest, WhenFinishingThenMaxAvailableSpaceIsNotExceeded) {
|
|
auto &commandStream = pCmdQ->getCS(1024);
|
|
auto &indirectHeap = pCmdQ->getIndirectHeap(IndirectHeap::Type::dynamicState, 10);
|
|
auto usedBeforeCS = commandStream.getUsed();
|
|
auto usedBeforeISH = indirectHeap.getUsed();
|
|
|
|
auto retVal = pCmdQ->finish();
|
|
|
|
auto usedAfterCS = commandStream.getUsed();
|
|
auto usedAfterISH = indirectHeap.getUsed();
|
|
EXPECT_LE(usedAfterCS - usedBeforeCS, commandStream.getMaxAvailableSpace());
|
|
EXPECT_LE(usedAfterISH - usedBeforeISH, indirectHeap.getMaxAvailableSpace());
|
|
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
}
|
|
|
|
HWTEST_P(OOMCommandQueueTest, WhenEnqueingMarkerThenMaxAvailableSpaceIsNotExceeded) {
|
|
auto &commandStream = pCmdQ->getCS(1024);
|
|
auto &indirectHeap = pCmdQ->getIndirectHeap(IndirectHeap::Type::dynamicState, 10);
|
|
auto usedBeforeCS = commandStream.getUsed();
|
|
auto usedBeforeISH = indirectHeap.getUsed();
|
|
|
|
Event event1(pCmdQ, CL_COMMAND_NDRANGE_KERNEL, 5, 15);
|
|
cl_event eventBeingWaitedOn = &event1;
|
|
cl_event eventReturned = nullptr;
|
|
auto retVal = pCmdQ->enqueueMarkerWithWaitList(
|
|
1,
|
|
&eventBeingWaitedOn,
|
|
&eventReturned);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
|
|
auto usedAfterCS = commandStream.getUsed();
|
|
auto usedAfterISH = indirectHeap.getUsed();
|
|
EXPECT_LE(usedAfterCS - usedBeforeCS, commandStream.getMaxAvailableSpace());
|
|
EXPECT_LE(usedAfterISH - usedBeforeISH, indirectHeap.getMaxAvailableSpace());
|
|
|
|
delete (Event *)eventReturned;
|
|
}
|
|
|
|
HWTEST_P(OOMCommandQueueTest, WhenEnqueingBarrierThenMaxAvailableSpaceIsNotExceeded) {
|
|
auto &commandStream = pCmdQ->getCS(1024);
|
|
auto &indirectHeap = pCmdQ->getIndirectHeap(IndirectHeap::Type::dynamicState, 10);
|
|
auto usedBeforeCS = commandStream.getUsed();
|
|
auto usedBeforeISH = indirectHeap.getUsed();
|
|
|
|
Event event1(pCmdQ, CL_COMMAND_NDRANGE_KERNEL, 5, 15);
|
|
cl_event eventBeingWaitedOn = &event1;
|
|
cl_event eventReturned = nullptr;
|
|
auto retVal = pCmdQ->enqueueBarrierWithWaitList(
|
|
1,
|
|
&eventBeingWaitedOn,
|
|
&eventReturned);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
|
|
auto usedAfterCS = commandStream.getUsed();
|
|
auto usedAfterISH = indirectHeap.getUsed();
|
|
EXPECT_LE(usedAfterCS - usedBeforeCS, commandStream.getMaxAvailableSpace());
|
|
EXPECT_LE(usedAfterISH - usedBeforeISH, indirectHeap.getMaxAvailableSpace());
|
|
|
|
delete (Event *)eventReturned;
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
OOM,
|
|
OOMCommandQueueTest,
|
|
testing::ValuesIn(oomSettings));
|