test: correct expectations in opencl tests if heapless enabled 2/n

Related-To: NEO-10641
Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
This commit is contained in:
Kamil Kopryk
2024-08-30 15:13:22 +00:00
committed by Compute-Runtime-Automation
parent 1aabc79176
commit d79889c3b2
12 changed files with 90 additions and 16 deletions

View File

@@ -38,6 +38,9 @@ TEST(CommandTest, GivenNoTerminateFlagWhenSubmittingMapUnmapThenCsrIsFlushed) {
CompletionStamp completionStamp = command->submit(20, false);
auto expectedTaskCount = initialTaskCount + 1;
if (csr.heaplessStateInitialized) {
expectedTaskCount++;
}
EXPECT_EQ(expectedTaskCount, completionStamp.taskCount);
}
@@ -71,7 +74,14 @@ TEST(CommandTest, GivenNoTerminateFlagWhenSubmittingMarkerThenCsrIsNotFlushed) {
std::unique_ptr<Command> command(new CommandWithoutKernel(*cmdQ));
CompletionStamp completionStamp = command->submit(20, false);
EXPECT_EQ(initialTaskCount, completionStamp.taskCount);
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());
}