Correct wrong constructor pattern in GraphicsAllocation

Related-To: NEO-6523
Signed-off-by: Krzysztof Gibala <[email protected]>

Author: kgibala

level_zero/core/test/unit_tests/gen9/test_cmdlist_create_gen9.cpp

@@ -57,9 +57,14 @@ class CommandListCreateGen9 : public DeviceFixture, public testing::Test {
         if (!buffer) {
             buffer = alignedMalloc(isaSize, 64);
         }
-        auto allocation = new NEO::GraphicsAllocation(0, NEO::AllocationType::INTERNAL_HOST_MEMORY,
-                                                      buffer, reinterpret_cast<uint64_t>(buffer), 0, isaSize,
-                                                      MemoryPool::System4KBPages);
+        auto allocation = new NEO::GraphicsAllocation(0,
+                                                      NEO::AllocationType::INTERNAL_HOST_MEMORY,
+                                                      buffer,
+                                                      reinterpret_cast<uint64_t>(buffer),
+                                                      0,
+                                                      isaSize,
+                                                      MemoryPool::System4KBPages,
+                                                      MemoryManager::maxOsContextCount);
         if (isaBuffer != nullptr) {
             memcpy_s(allocation->getUnderlyingBuffer(), allocation->getUnderlyingBufferSize(), isaBuffer, isaSize);
         }

level_zero/core/test/unit_tests/mocks/mock_cmdlist.cpp

@@ -18,9 +18,14 @@ MockCommandList::MockCommandList(Device *device) : WhiteBox<::L0::CommandList>(d
     this->device = device;
     size_t batchBufferSize = 65536u;
     batchBuffer = new uint8_t[batchBufferSize];
-    mockAllocation = new NEO::GraphicsAllocation(0, NEO::AllocationType::INTERNAL_HOST_MEMORY,
-                                                 &batchBuffer, reinterpret_cast<uint64_t>(&batchBuffer), 0, sizeof(batchBufferSize),
-                                                 MemoryPool::System4KBPages);
+    mockAllocation = new NEO::GraphicsAllocation(0,
+                                                 NEO::AllocationType::INTERNAL_HOST_MEMORY,
+                                                 &batchBuffer,
+                                                 reinterpret_cast<uint64_t>(&batchBuffer),
+                                                 0,
+                                                 sizeof(batchBufferSize),
+                                                 MemoryPool::System4KBPages,
+                                                 NEO::MemoryManager::maxOsContextCount);
 }
 
 MockCommandList::~MockCommandList() {

level_zero/core/test/unit_tests/mocks/mock_event.cpp

@@ -10,9 +10,14 @@
 namespace L0 {
 namespace ult {
 
-Mock<Event>::Mock() : mockAllocation(0, NEO::AllocationType::INTERNAL_HOST_MEMORY,
-                                     &memory, reinterpret_cast<uint64_t>(&memory), 0, sizeof(memory),
-                                     MemoryPool::System4KBPages) {}
+Mock<Event>::Mock() : mockAllocation(0,
+                                     NEO::AllocationType::INTERNAL_HOST_MEMORY,
+                                     &memory,
+                                     reinterpret_cast<uint64_t>(&memory),
+                                     0,
+                                     sizeof(memory),
+                                     MemoryPool::System4KBPages,
+                                     NEO::MemoryManager::maxOsContextCount) {}
 
 Mock<Event>::~Mock() {}
 

level_zero/core/test/unit_tests/mocks/mock_event.h

@@ -68,9 +68,14 @@ class MockEvent : public ::L0::Event {
   public:
     using ::L0::Event::l3FlushAppliedOnKernel;
     MockEvent() {
-        mockAllocation.reset(new NEO::MockGraphicsAllocation(0, NEO::AllocationType::INTERNAL_HOST_MEMORY,
-                                                             reinterpret_cast<void *>(0x1234), 0x1000, 0, sizeof(uint32_t),
-                                                             MemoryPool::System4KBPages));
+        mockAllocation.reset(new NEO::MockGraphicsAllocation(0,
+                                                             NEO::AllocationType::INTERNAL_HOST_MEMORY,
+                                                             reinterpret_cast<void *>(0x1234),
+                                                             0x1000,
+                                                             0,
+                                                             sizeof(uint32_t),
+                                                             MemoryPool::System4KBPages,
+                                                             NEO::MemoryManager::maxOsContextCount));
         this->timestampSizeInDw = 1;
         this->contextStartOffset = 0;
         this->contextEndOffset = 4;

level_zero/core/test/unit_tests/mocks/mock_kernel.h

@@ -115,9 +115,14 @@ struct Mock<::L0::Kernel> : public WhiteBox<::L0::Kernel> {
 
         this->kernelImmData = &immutableData;
 
-        auto allocation = new NEO::GraphicsAllocation(0, NEO::AllocationType::KERNEL_ISA,
-                                                      nullptr, 0, 0, 4096,
-                                                      MemoryPool::System4KBPages);
+        auto allocation = new NEO::GraphicsAllocation(0,
+                                                      NEO::AllocationType::KERNEL_ISA,
+                                                      nullptr,
+                                                      0,
+                                                      0,
+                                                      4096,
+                                                      MemoryPool::System4KBPages,
+                                                      NEO::MemoryManager::maxOsContextCount);
 
         immutableData.isaGraphicsAllocation.reset(allocation);