feature: multi-gpu madvise

Related-To: NEO-16389

Signed-off-by: Chandio, Bibrak Qamar <[email protected]>

Author: bibrak

shared/source/os_interface/linux/xe/ioctl_helper_xe.cpp

@@ -806,12 +806,20 @@ uint64_t IoctlHelperXe::getPreferredLocationArgs(MemAdvise memAdviseOp) {
         // Assumes that the default location is Device VRAM.
         const auto preferredLocation = static_cast<uint64_t>(getDrmParamValue(DrmParam::memoryAdviseLocationDevice));
         const auto policy = static_cast<uint64_t>(getDrmParamValue(DrmParam::memoryAdviseMigrationPolicyAllPages));
-        param = (preferredLocation << 32) | policy;
+        const auto regionInstance = static_cast<uint64_t>(0);
+        param = (preferredLocation << 32) | (policy << 16) | regionInstance;
+    } break;
+    case MemAdvise::setPreferredLocation: {
+        const auto preferredLocation = static_cast<uint64_t>(deviceFd);
+        const auto policy = static_cast<uint64_t>(getDrmParamValue(DrmParam::memoryAdviseMigrationPolicyAllPages));
+        const auto regionInstance = static_cast<uint64_t>(1);
+        param = (preferredLocation << 32) | (policy << 16) | regionInstance;
     } break;
     case MemAdvise::setSystemMemoryPreferredLocation: {
         const auto preferredLocation = static_cast<uint64_t>(getDrmParamValue(DrmParam::memoryAdviseLocationSystem));
         const auto policy = static_cast<uint64_t>(getDrmParamValue(DrmParam::memoryAdviseMigrationPolicySystemPages));
-        param = (preferredLocation << 32) | policy;
+        const auto regionInstance = static_cast<uint64_t>(0);
+        param = (preferredLocation << 32) | (policy << 16) | regionInstance;
     } break;
     default:
         xeLog(" Invalid advise operation %s\n", __FUNCTION__);