refactor: Use helper to check platform support for KMD migration

Related-To: NEO-6465

Signed-off-by: Milczarek, Slawomir <[email protected]>

Author: smilczar

shared/source/os_interface/linux/drm_allocation.cpp

@@ -8,6 +8,7 @@
 #include "shared/source/os_interface/linux/drm_allocation.h"
 
 #include "shared/source/debug_settings/debug_settings_manager.h"
+#include "shared/source/execution_environment/root_device_environment.h"
 #include "shared/source/helpers/basic_math.h"
 #include "shared/source/memory_manager/residency.h"
 #include "shared/source/os_interface/linux/cache_info.h"
@@ -17,6 +18,7 @@
 #include "shared/source/os_interface/linux/ioctl_helper.h"
 #include "shared/source/os_interface/linux/os_context_linux.h"
 #include "shared/source/os_interface/os_context.h"
+#include "shared/source/os_interface/product_helper.h"
 
 #include <sstream>
 
@@ -334,9 +336,12 @@ bool DrmAllocation::shouldAllocationPageFault(const Drm *drm) {
         return DebugManager.flags.EnableImplicitMigrationOnFaultableHardware.get();
     }
 
+    auto &productHelper = drm->getRootDeviceEnvironment().getHelper<ProductHelper>();
+    auto isKmdMigrationSupported = productHelper.isKmdMigrationSupported();
+
     switch (this->allocationType) {
     case AllocationType::UNIFIED_SHARED_MEMORY:
-        return DebugManager.flags.UseKmdMigration.get() > 0;
+        return (DebugManager.flags.UseKmdMigration.get() == -1) ? isKmdMigrationSupported : DebugManager.flags.UseKmdMigration.get();
     case AllocationType::BUFFER:
         return DebugManager.flags.UseKmdMigrationForBuffers.get() > 0;
     default:

shared/test/unit_test/os_interface/linux/drm_vm_bind_prelim_tests.cpp

@@ -8,6 +8,7 @@
 #include "shared/source/execution_environment/execution_environment.h"
 #include "shared/source/execution_environment/root_device_environment.h"
 #include "shared/source/os_interface/linux/os_context_linux.h"
+#include "shared/source/os_interface/product_helper.h"
 #include "shared/test/common/helpers/debug_manager_state_restore.h"
 #include "shared/test/common/helpers/engine_descriptor_helper.h"
 #include "shared/test/common/libult/linux/drm_query_mock.h"
@@ -121,3 +122,27 @@ TEST(DrmVmBindTest, givenUseKmdMigrationWhenCallingBindBoOnUnifiedSharedMemoryTh
     EXPECT_FALSE(bo.isExplicitResidencyRequired());
     EXPECT_EQ(DrmPrelimHelper::getImmediateVmBindFlag(), drm.context.receivedVmBind->flags);
 }
+
+TEST(DrmVmBindTest, givenDefaultDriverSettingsWhenCallingBindBoOnUnifiedSharedMemoryThenMarkAllocationShouldPageFaultWhenKmdMigrationIsSupported) {
+    auto executionEnvironment = std::make_unique<MockExecutionEnvironment>();
+    executionEnvironment->rootDeviceEnvironments[0]->initGmm();
+    executionEnvironment->initializeMemoryManager();
+
+    DrmQueryMock drm(*executionEnvironment->rootDeviceEnvironments[0]);
+    drm.pageFaultSupported = true;
+
+    OsContextLinux osContext(drm, 0, 0u, EngineDescriptorHelper::getDefaultDescriptor());
+    osContext.ensureContextInitialized();
+    uint32_t vmHandleId = 0;
+
+    MockBufferObject bo(0u, &drm, 3, 0, 0, 1);
+    MockDrmAllocation allocation(0u, AllocationType::UNIFIED_SHARED_MEMORY, MemoryPool::LocalMemory);
+    allocation.bufferObjects[0] = &bo;
+
+    allocation.bindBO(&bo, &osContext, vmHandleId, nullptr, true);
+
+    auto &productHelper = drm.getRootDeviceEnvironment().getHelper<ProductHelper>();
+    auto isKmdMigrationSupported = productHelper.isKmdMigrationSupported();
+
+    EXPECT_EQ(isKmdMigrationSupported, allocation.shouldAllocationPageFault(&drm));
+}