107 lines
4.7 KiB
C++
107 lines
4.7 KiB
C++
/*
|
|
* Copyright (C) 2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/command_stream/preemption.h"
|
|
#include "shared/test/unit_test/cmd_parse/hw_parse.h"
|
|
#include "shared/test/unit_test/mocks/mock_command_stream_receiver.h"
|
|
|
|
#include "opencl/source/event/user_event.h"
|
|
#include "opencl/test/unit_test/fixtures/cl_preemption_fixture.h"
|
|
#include "opencl/test/unit_test/libult/ult_command_stream_receiver.h"
|
|
|
|
using namespace NEO;
|
|
|
|
using Gen8PreemptionEnqueueKernelTest = PreemptionEnqueueKernelTest;
|
|
using Gen8ClPreemptionTests = DevicePreemptionTests;
|
|
|
|
GEN8TEST_F(Gen8ClPreemptionTests, allowThreadGroupPreemptionReturnsTrue) {
|
|
PreemptionFlags flags = {};
|
|
PreemptionHelper::setPreemptionLevelFlags(flags, device->getDevice(), kernel.get());
|
|
EXPECT_TRUE(PreemptionHelper::allowThreadGroupPreemption(flags));
|
|
}
|
|
|
|
GEN8TEST_F(Gen8PreemptionEnqueueKernelTest, givenSecondEnqueueWithTheSamePreemptionRequestThenDontReprogram) {
|
|
pDevice->setPreemptionMode(PreemptionMode::ThreadGroup);
|
|
auto &csr = pDevice->getUltCommandStreamReceiver<FamilyType>();
|
|
csr.getMemoryManager()->setForce32BitAllocations(false);
|
|
csr.setMediaVFEStateDirty(false);
|
|
size_t off[3] = {0, 0, 0};
|
|
size_t gws[3] = {1, 1, 1};
|
|
|
|
MockKernelWithInternals mockKernel(*pClDevice);
|
|
|
|
HardwareParse hwParser;
|
|
pCmdQ->enqueueKernel(mockKernel.mockKernel, 1, off, gws, nullptr, 0, nullptr, nullptr);
|
|
hwParser.parseCommands<FamilyType>(csr.commandStream);
|
|
auto offset = csr.commandStream.getUsed();
|
|
pCmdQ->enqueueKernel(mockKernel.mockKernel, 1, off, gws, nullptr, 0, nullptr, nullptr);
|
|
pCmdQ->flush();
|
|
hwParser.parseCommands<FamilyType>(csr.commandStream, offset);
|
|
|
|
size_t numMmiosFound = countMmio<FamilyType>(hwParser.cmdList.begin(), hwParser.cmdList.end(), 0x2248u);
|
|
EXPECT_EQ(1U, numMmiosFound);
|
|
}
|
|
|
|
GEN8TEST_F(Gen8PreemptionEnqueueKernelTest, givenValidKernelForPreemptionWhenEnqueueKernelCalledThenPassDevicePreemptionMode) {
|
|
pDevice->setPreemptionMode(PreemptionMode::ThreadGroup);
|
|
auto mockCsr = new MockCsrHw2<FamilyType>(*pDevice->executionEnvironment, pDevice->getRootDeviceIndex());
|
|
pDevice->resetCommandStreamReceiver(mockCsr);
|
|
|
|
MockKernelWithInternals mockKernel(*pClDevice);
|
|
PreemptionFlags flags = {};
|
|
MultiDispatchInfo multiDispatch(mockKernel.mockKernel);
|
|
EXPECT_EQ(PreemptionMode::ThreadGroup, PreemptionHelper::taskPreemptionMode(*pDevice, multiDispatch));
|
|
|
|
size_t gws[3] = {1, 0, 0};
|
|
pCmdQ->enqueueKernel(mockKernel.mockKernel, 1, nullptr, gws, nullptr, 0, nullptr, nullptr);
|
|
pCmdQ->flush();
|
|
|
|
EXPECT_EQ(1, mockCsr->flushCalledCount);
|
|
EXPECT_EQ(PreemptionMode::ThreadGroup, mockCsr->passedDispatchFlags.preemptionMode);
|
|
}
|
|
|
|
GEN8TEST_F(Gen8PreemptionEnqueueKernelTest, givenValidKernelForPreemptionWhenEnqueueKernelCalledAndBlockedThenPassDevicePreemptionMode) {
|
|
pDevice->setPreemptionMode(PreemptionMode::ThreadGroup);
|
|
auto mockCsr = new MockCsrHw2<FamilyType>(*pDevice->executionEnvironment, pDevice->getRootDeviceIndex());
|
|
pDevice->resetCommandStreamReceiver(mockCsr);
|
|
|
|
MockKernelWithInternals mockKernel(*pClDevice);
|
|
PreemptionFlags flags = {};
|
|
PreemptionHelper::setPreemptionLevelFlags(flags, *pDevice, mockKernel.mockKernel);
|
|
EXPECT_EQ(PreemptionMode::ThreadGroup, PreemptionHelper::taskPreemptionMode(pDevice->getPreemptionMode(), flags));
|
|
|
|
UserEvent userEventObj;
|
|
cl_event userEvent = &userEventObj;
|
|
size_t gws[3] = {1, 0, 0};
|
|
pCmdQ->enqueueKernel(mockKernel.mockKernel, 1, nullptr, gws, nullptr, 1, &userEvent, nullptr);
|
|
pCmdQ->flush();
|
|
EXPECT_EQ(0, mockCsr->flushCalledCount);
|
|
|
|
userEventObj.setStatus(CL_COMPLETE);
|
|
pCmdQ->flush();
|
|
EXPECT_EQ(1, mockCsr->flushCalledCount);
|
|
EXPECT_EQ(PreemptionMode::ThreadGroup, mockCsr->passedDispatchFlags.preemptionMode);
|
|
}
|
|
|
|
GEN8TEST_F(Gen8PreemptionEnqueueKernelTest, givenDisabledPreemptionWhenEnqueueKernelCalledThenPassDisabledPreemptionMode) {
|
|
pDevice->setPreemptionMode(PreemptionMode::Disabled);
|
|
auto mockCsr = new MockCsrHw2<FamilyType>(*pDevice->executionEnvironment, pDevice->getRootDeviceIndex());
|
|
pDevice->resetCommandStreamReceiver(mockCsr);
|
|
|
|
MockKernelWithInternals mockKernel(*pClDevice);
|
|
PreemptionFlags flags = {};
|
|
PreemptionHelper::setPreemptionLevelFlags(flags, *pDevice, mockKernel.mockKernel);
|
|
EXPECT_EQ(PreemptionMode::Disabled, PreemptionHelper::taskPreemptionMode(pDevice->getPreemptionMode(), flags));
|
|
|
|
size_t gws[3] = {1, 0, 0};
|
|
pCmdQ->enqueueKernel(mockKernel.mockKernel, 1, nullptr, gws, nullptr, 0, nullptr, nullptr);
|
|
pCmdQ->flush();
|
|
|
|
EXPECT_EQ(1, mockCsr->flushCalledCount);
|
|
EXPECT_EQ(PreemptionMode::Disabled, mockCsr->passedDispatchFlags.preemptionMode);
|
|
}
|