Move OsContext id setting to constructor.

Change-Id: I1b809befc02536257800e3667307b8deabd5c95d

Author: MichalMrozek

runtime/device/device.cpp

@@ -131,7 +131,7 @@ bool Device::createDeviceImpl(const HardwareInfo *pHwInfo, Device &outDevice) {
     }
     executionEnvironment->initializeMemoryManager(outDevice.getEnabled64kbPages(), outDevice.getHardwareCapabilities().localMemorySupported, outDevice.getDeviceIndex());
 
-    outDevice.osContext = new OsContext(executionEnvironment->osInterface.get());
+    outDevice.osContext = new OsContext(executionEnvironment->osInterface.get(), outDevice.getDeviceIndex());
     executionEnvironment->memoryManager->registerOsContext(outDevice.osContext);
 
     outDevice.commandStreamReceiver = executionEnvironment->commandStreamReceivers[outDevice.getDeviceIndex()].get();

runtime/memory_manager/memory_manager.cpp

@@ -387,9 +387,12 @@ RequirementsStatus MemoryManager::checkAllocationsForOverlapping(AllocationRequi
 }
 
 void MemoryManager::registerOsContext(OsContext *contextToRegister) {
+    auto contextId = contextToRegister->getContextId();
+    if (contextId + 1 > registeredOsContexts.size()) {
+        registeredOsContexts.resize(contextId + 1);
+    }
     contextToRegister->incRefInternal();
-    contextToRegister->setContextId(static_cast<uint32_t>(registeredOsContexts.size()));
-    registeredOsContexts.push_back(contextToRegister);
+    registeredOsContexts[contextToRegister->getContextId()] = contextToRegister;
 }
 
 bool MemoryManager::getAllocationData(AllocationData &allocationData, bool allocateMemory, const void *hostPtr, size_t size, GraphicsAllocation::AllocationType type) {

runtime/os_interface/linux/os_context_linux.cpp

@@ -24,7 +24,7 @@
 
 namespace OCLRT {
 class OsContext::OsContextImpl {};
-OsContext::OsContext(OSInterface *osInterface) {
+OsContext::OsContext(OSInterface *osInterface, uint32_t contextId) : contextId(contextId) {
     osContextImpl = std::make_unique<OsContext::OsContextImpl>();
 }
 

runtime/os_interface/os_context.h

@@ -29,13 +29,12 @@ class OSInterface;
 class OsContext : public ReferenceTrackedObject<OsContext> {
   public:
     class OsContextImpl;
-    OsContext(OSInterface *osInterface);
+    OsContext(OSInterface *osInterface, uint32_t contextId);
     ~OsContext() override;
     OsContextImpl *get() const {
         return osContextImpl.get();
     };
 
-    void setContextId(uint32_t inputContextId) { contextId = inputContextId; }
     uint32_t getContextId() { return contextId; }
 
   protected:

runtime/os_interface/windows/os_context_win.cpp

@@ -53,7 +53,7 @@ void OsContextWin::resetMonitoredFenceParams(D3DKMT_HANDLE &handle, uint64_t *cp
     monitoredFence.gpuAddress = gpuAddress;
 }
 
-OsContext::OsContext(OSInterface *osInterface) {
+OsContext::OsContext(OSInterface *osInterface, uint32_t contextId) : contextId(contextId) {
     if (osInterface) {
         osContextImpl = std::make_unique<OsContextWin>(*osInterface->get()->getWddm());
     }

unit_tests/command_stream/command_stream_receiver_with_aub_dump_tests.cpp

@@ -147,7 +147,7 @@ HWTEST_P(CommandStreamReceiverWithAubDumpTest, givenCommandStreamReceiverWithAub
     BatchBuffer batchBuffer{cs.getGraphicsAllocation(), 0, 0, nullptr, false, false, QueueThrottle::MEDIUM, cs.getUsed(), &cs};
     auto engineType = OCLRT::ENGINE_RCS;
 
-    OsContext osContext(nullptr);
+    OsContext osContext(nullptr, 0u);
 
     ResidencyContainer allocationsForResidency;
     FlushStamp flushStamp = csrWithAubDump->flush(batchBuffer, engineType, &allocationsForResidency, osContext);
@@ -190,7 +190,7 @@ HWTEST_P(CommandStreamReceiverWithAubDumpTest, givenCommandStreamReceiverWithAub
     ASSERT_NE(nullptr, gfxAllocation);
 
     ResidencyContainer allocationsForResidency = {gfxAllocation};
-    OsContext osContext(nullptr);
+    OsContext osContext(nullptr, 0u);
     csrWithAubDump->processResidency(&allocationsForResidency, osContext);
 
     EXPECT_TRUE(csrWithAubDump->processResidencyParameterization.wasCalled);

unit_tests/memory_manager/memory_manager_tests.cpp

@@ -1825,23 +1825,33 @@ TEST(GraphicsAllocation, givenSharedHandleBasedConstructorWhenGraphicsAllocation
 }
 
 TEST(ResidencyDataTest, givenOsContextWhenItIsRegisteredToMemoryManagerThenRefCountIncreases) {
-    auto osContext = new OsContext(nullptr);
+    auto osContext = new OsContext(nullptr, 0u);
     OsAgnosticMemoryManager memoryManager;
     memoryManager.registerOsContext(osContext);
     EXPECT_EQ(1u, memoryManager.getOsContextCount());
     EXPECT_EQ(1, osContext->getRefInternalCount());
 }
 
+TEST(ResidencyDataTest, givenTwoOsContextsWhenTheyAreRegistredFromHigherToLowerThenProperSizeIsReturned) {
+    auto osContext2 = new OsContext(nullptr, 1u);
+    auto osContext = new OsContext(nullptr, 0u);
+    OsAgnosticMemoryManager memoryManager;
+    memoryManager.registerOsContext(osContext2);
+    memoryManager.registerOsContext(osContext);
+    EXPECT_EQ(2u, memoryManager.getOsContextCount());
+    EXPECT_EQ(1, osContext->getRefInternalCount());
+    EXPECT_EQ(1, osContext2->getRefInternalCount());
+}
+
 TEST(ResidencyDataTest, givenResidencyDataWhenUpdateCompletionDataIsCalledThenItIsProperlyUpdated) {
     struct mockResidencyData : public ResidencyData {
         using ResidencyData::completionData;
     };
 
     mockResidencyData residency;
 
-    OsContext osContext(nullptr);
-    OsContext osContext2(nullptr);
-    osContext2.setContextId(1u);
+    OsContext osContext(nullptr, 0u);
+    OsContext osContext2(nullptr, 1u);
 
     auto lastFenceValue = 45llu;
     auto lastFenceValue2 = 23llu;

unit_tests/os_interface/linux/drm_command_stream_tests.cpp

@@ -57,7 +57,7 @@ class DrmCommandStreamFixture {
     const int mockFd = 33;
 
     void SetUp() {
-        osContext = std::make_unique<OsContext>(nullptr);
+        osContext = std::make_unique<OsContext>(nullptr, 0u);
         this->dbgState = new DebugManagerStateRestore();
         //make sure this is disabled, we don't want test this now
         DebugManager.flags.EnableForcePin.set(false);

unit_tests/os_interface/linux/os_interface_linux_tests.cpp

@@ -37,7 +37,7 @@ TEST(OsInterfaceTest, GivenLinuxOsInterfaceWhenDeviceHandleQueriedthenZeroIsRetu
 
 TEST(OsContextTest, WhenOsContextIsCreatedThenImplIsAvailable) {
     OSInterface osInterface;
-    auto osContext = std::make_unique<OsContext>(&osInterface);
+    auto osContext = std::make_unique<OsContext>(&osInterface, 0u);
     EXPECT_NE(nullptr, osContext->get());
 }
-} // namespace OCLRT
+} // namespace OCLRT

unit_tests/os_interface/windows/device_command_stream_tests.cpp

@@ -279,7 +279,7 @@ TEST(WddmPreemptionHeaderTests, givenWddmCommandStreamReceiverWhenPreemptionIsOf
     LinearStream cs(commandBuffer);
 
     BatchBuffer batchBuffer{cs.getGraphicsAllocation(), 0, 0, nullptr, false, false, QueueThrottle::MEDIUM, cs.getUsed(), &cs};
-    OsContext *osContext = new OsContext(executionEnvironment.osInterface.get());
+    OsContext *osContext = new OsContext(executionEnvironment.osInterface.get(), 0u);
     osContext->incRefInternal();
     executionEnvironment.commandStreamReceivers[0u]->flush(batchBuffer, EngineType::ENGINE_RCS, &executionEnvironment.commandStreamReceivers[0u]->getResidencyAllocations(), *osContext);
     auto commandHeader = wddm->submitResult.commandHeaderSubmitted;
@@ -306,7 +306,7 @@ TEST(WddmPreemptionHeaderTests, givenWddmCommandStreamReceiverWhenPreemptionIsOn
     LinearStream cs(commandBuffer);
 
     BatchBuffer batchBuffer{cs.getGraphicsAllocation(), 0, 0, nullptr, false, false, QueueThrottle::MEDIUM, cs.getUsed(), &cs};
-    OsContext *osContext = new OsContext(executionEnvironment.osInterface.get());
+    OsContext *osContext = new OsContext(executionEnvironment.osInterface.get(), 0u);
     osContext->incRefInternal();
     executionEnvironment.commandStreamReceivers[0u]->flush(batchBuffer, EngineType::ENGINE_RCS, &executionEnvironment.commandStreamReceivers[0u]->getResidencyAllocations(), *osContext);
     auto commandHeader = wddm->submitResult.commandHeaderSubmitted;