test: correct expectations in opencl tests if heapless enabled 4/4

Related-To: NEO-10641
Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
This commit is contained in:
Kamil Kopryk
2024-09-06 06:21:45 +00:00
committed by Compute-Runtime-Automation
parent 41e24635e3
commit c7a6a74e4e
32 changed files with 389 additions and 150 deletions

View File

@@ -27,10 +27,9 @@ using namespace NEO;
TEST(CommandTest, GivenNoTerminateFlagWhenSubmittingMapUnmapThenCsrIsFlushed) {
auto device = std::make_unique<MockClDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(nullptr));
std::unique_ptr<MockCommandQueue> cmdQ(new MockCommandQueue(nullptr, device.get(), nullptr, false));
MockCommandStreamReceiver csr(*device->getExecutionEnvironment(), device->getRootDeviceIndex(), device->getDeviceBitfield());
MockBuffer buffer;
auto initialTaskCount = csr.peekTaskCount();
MockBuffer buffer;
auto initialTaskCount = 0u;
MemObjSizeArray size = {{1, 1, 1}};
MemObjOffsetArray offset = {{0, 0, 0}};
@@ -38,7 +37,7 @@ TEST(CommandTest, GivenNoTerminateFlagWhenSubmittingMapUnmapThenCsrIsFlushed) {
CompletionStamp completionStamp = command->submit(20, false);
auto expectedTaskCount = initialTaskCount + 1;
if (csr.heaplessStateInitialized) {
if (cmdQ->heaplessStateInitEnabled) {
expectedTaskCount++;
}
EXPECT_EQ(expectedTaskCount, completionStamp.taskCount);
@@ -67,22 +66,15 @@ TEST(CommandTest, GivenTerminateFlagWhenSubmittingMapUnmapThenFlushIsAborted) {
TEST(CommandTest, GivenNoTerminateFlagWhenSubmittingMarkerThenCsrIsNotFlushed) {
auto device = std::make_unique<MockClDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(nullptr));
std::unique_ptr<MockCommandQueue> cmdQ(new MockCommandQueue(nullptr, device.get(), nullptr, false));
MockCommandStreamReceiver csr(*device->getExecutionEnvironment(), device->getRootDeviceIndex(), device->getDeviceBitfield());
MockBuffer buffer;
auto initialTaskCount = csr.peekTaskCount();
auto heaplessStateInit = cmdQ->getHeaplessStateInitEnabled();
auto initialTaskCount = heaplessStateInit ? 1u : 0u;
std::unique_ptr<Command> command(new CommandWithoutKernel(*cmdQ));
CompletionStamp completionStamp = command->submit(20, false);
auto heaplessStateInit = cmdQ->getHeaplessStateInitEnabled();
if (heaplessStateInit) {
EXPECT_EQ(1u, initialTaskCount);
} else {
EXPECT_EQ(0u, initialTaskCount);
}
EXPECT_EQ(0u, completionStamp.taskCount);
EXPECT_EQ(initialTaskCount, csr.peekTaskCount());
EXPECT_EQ(initialTaskCount, completionStamp.taskCount);
}
TEST(CommandTest, GivenTerminateFlagWhenSubmittingMarkerThenFlushIsAborted) {