Add missing getter methods

Signed-off-by: Maciej Dziuban <[email protected]>

Author: DziubanMaciejIntel

opencl/test/unit_test/command_stream/command_stream_receiver_tests.cpp

@@ -1147,19 +1147,28 @@ HWTEST_F(CommandStreamReceiverTest, whenCreatingCommandStreamReceiverThenLastAdd
 
 HWTEST_F(CommandStreamReceiverTest, givenDebugFlagWhenCreatingCsrThenSetEnableStaticPartitioningAccordingly) {
     DebugManagerStateRestore restore{};
+
     {
-        MockDevice device{};
-        EXPECT_FALSE(device.getUltCommandStreamReceiver<FamilyType>().staticWorkPartitioningEnabled);
+        UltDeviceFactory deviceFactory{1, 2};
+        MockDevice &device = *deviceFactory.rootDevices[0];
+        EXPECT_FALSE(device.getGpgpuCommandStreamReceiver().isStaticWorkPartitioningEnabled());
+        EXPECT_EQ(0u, device.getGpgpuCommandStreamReceiver().getWorkPartitionAllocationGpuAddress());
     }
     {
         DebugManager.flags.EnableStaticPartitioning.set(0);
-        MockDevice device{};
-        EXPECT_FALSE(device.getUltCommandStreamReceiver<FamilyType>().staticWorkPartitioningEnabled);
+        UltDeviceFactory deviceFactory{1, 2};
+        MockDevice &device = *deviceFactory.rootDevices[0];
+        EXPECT_FALSE(device.getGpgpuCommandStreamReceiver().isStaticWorkPartitioningEnabled());
+        EXPECT_EQ(0u, device.getGpgpuCommandStreamReceiver().getWorkPartitionAllocationGpuAddress());
     }
     {
         DebugManager.flags.EnableStaticPartitioning.set(1);
-        MockDevice device{};
-        EXPECT_TRUE(device.getUltCommandStreamReceiver<FamilyType>().staticWorkPartitioningEnabled);
+        UltDeviceFactory deviceFactory{1, 2};
+        MockDevice &device = *deviceFactory.rootDevices[0];
+        EXPECT_TRUE(device.getGpgpuCommandStreamReceiver().isStaticWorkPartitioningEnabled());
+        EXPECT_NE(0u, device.getGpgpuCommandStreamReceiver().getWorkPartitionAllocationGpuAddress());
+        const auto gpuVa = device.getGpgpuCommandStreamReceiver().getWorkPartitionAllocation()->getGpuAddress();
+        EXPECT_EQ(gpuVa, device.getGpgpuCommandStreamReceiver().getWorkPartitionAllocationGpuAddress());
     }
 }
 

shared/source/command_stream/command_stream_receiver.cpp

@@ -187,6 +187,13 @@ OSInterface *CommandStreamReceiver::getOSInterface() const {
     return executionEnvironment.rootDeviceEnvironments[rootDeviceIndex]->osInterface.get();
 }
 
+uint64_t CommandStreamReceiver::getWorkPartitionAllocationGpuAddress() const {
+    if (isStaticWorkPartitioningEnabled()) {
+        return getWorkPartitionAllocation()->getGpuAddress();
+    }
+    return 0;
+}
+
 bool CommandStreamReceiver::isRcs() const {
     return this->osContext->getEngineType() == aub_stream::ENGINE_RCS;
 }

shared/source/command_stream/command_stream_receiver.h

@@ -228,6 +228,12 @@ class CommandStreamReceiver {
         return false;
     }
 
+    bool isStaticWorkPartitioningEnabled() const {
+        return staticWorkPartitioningEnabled;
+    }
+
+    uint64_t getWorkPartitionAllocationGpuAddress() const;
+
     bool isRcs() const;
 
     virtual void initializeDefaultsForInternalEngine(){};