Add helpers for page fault support

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

Author: smilczar

opencl/test/unit_test/os_interface/linux/drm_buffer_object_tests.cpp

@@ -532,3 +532,14 @@ TEST_F(DrmBufferObjectTest, given48bitAddressWhenSetThenAddressIsCanonized) {
     auto boAddress = bo.peekAddress();
     EXPECT_EQ(boAddress, expectedAddress);
 }
+
+TEST_F(DrmBufferObjectTest, whenBoRequiresExplicitResidencyThenTheCorrespondingQueryReturnsCorrectValue) {
+    MockExecutionEnvironment executionEnvironment(defaultHwInfo.get());
+    DrmMock drm(*(executionEnvironment.rootDeviceEnvironments[0].get()));
+    MockBufferObject bo(&drm, 0, 0, 1);
+
+    for (auto required : {false, true}) {
+        bo.requireExplicitResidency(required);
+        EXPECT_EQ(required, bo.isExplicitResidencyRequired());
+    }
+}

shared/source/os_interface/linux/drm_allocation.cpp

@@ -138,6 +138,7 @@ void DrmAllocation::registerBOBindExtHandle(Drm *drm) {
                 }
 
                 bo->requireImmediateBinding(true);
+                bo->requireExplicitResidency(!shouldAllocationPageFault(drm));
             }
         }
     }

shared/source/os_interface/linux/drm_allocation.h

@@ -88,6 +88,7 @@ class DrmAllocation : public GraphicsAllocation {
     void freeRegisteredBOBindExtHandles(Drm *drm);
     void linkWithRegisteredHandle(uint32_t handle);
     MOCKABLE_VIRTUAL void markForCapture();
+    bool shouldAllocationPageFault(Drm *drm);
 
   protected:
     BufferObjects bufferObjects{};

shared/source/os_interface/linux/drm_allocation_extended.cpp

@@ -42,4 +42,8 @@ bool DrmAllocation::setMemAdvise(Drm *drm, MemAdviseFlags flags) {
     return true;
 }
 
+bool DrmAllocation::shouldAllocationPageFault(Drm *drm) {
+    return false;
+}
+
 } // namespace NEO

shared/source/os_interface/linux/drm_buffer_object.h

@@ -91,6 +91,13 @@ class BufferObject {
         requiresImmediateBinding = required;
     }
 
+    bool isExplicitResidencyRequired() {
+        return requiresExplicitResidency;
+    }
+    void requireExplicitResidency(bool required) {
+        requiresExplicitResidency = required;
+    }
+
     void setCacheRegion(CacheRegion regionIndex) { cacheRegion = regionIndex; }
     CacheRegion peekCacheRegion() const { return cacheRegion; }
 
@@ -133,6 +140,7 @@ class BufferObject {
     uint32_t tiling_mode;
     bool allowCapture = false;
     bool requiresImmediateBinding = false;
+    bool requiresExplicitResidency = false;
 
     uint32_t getOsContextId(OsContext *osContext);
     MOCKABLE_VIRTUAL void fillExecObject(drm_i915_gem_exec_object2 &execObject, OsContext *osContext, uint32_t vmHandleId, uint32_t drmContextId);

shared/source/os_interface/linux/drm_neo.h

@@ -172,6 +172,9 @@ class Drm : public DriverModel {
     MOCKABLE_VIRTUAL bool isVmBindAvailable();
     MOCKABLE_VIRTUAL bool registerResourceClasses();
 
+    void queryPageFaultSupport();
+    bool hasPageFaultSupport();
+
     MOCKABLE_VIRTUAL uint32_t registerResource(ResourceClass classType, const void *data, size_t size);
     MOCKABLE_VIRTUAL void unregisterResource(uint32_t handle);
     MOCKABLE_VIRTUAL uint32_t registerIsaCookie(uint32_t isaHandle);
@@ -323,6 +326,7 @@ class Drm : public DriverModel {
     bool bindAvailable = false;
     bool directSubmissionActive = false;
     bool contextDebugSupported = false;
+    bool pageFaultSupported = false;
 
   private:
     int getParamIoctl(int param, int *dstValue);

shared/source/os_interface/linux/drm_query_extended.cpp

@@ -71,4 +71,11 @@ uint32_t Drm::createDrmContextExt(drm_i915_gem_context_create_ext &gcc, uint32_t
 void Drm::appendDrmContextFlags(drm_i915_gem_context_create_ext &gcc, bool isSpecialContextRequested) {
 }
 
+void Drm::queryPageFaultSupport() {
+}
+
+bool Drm::hasPageFaultSupport() {
+    return false;
+}
+
 } // namespace NEO

shared/test/common/os_interface/linux/drm_neo_create.cpp

@@ -52,6 +52,9 @@ Drm *Drm::create(std::unique_ptr<HwDeviceIdDrm> &&hwDeviceId, RootDeviceEnvironm
     if (!drm->isPerContextVMRequired()) {
         drm->createVirtualMemoryAddressSpace(HwHelper::getSubDevicesCount(hwInfo));
     }
+
+    drm->queryPageFaultSupport();
+
     return drm;
 }
 

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

@@ -11,6 +11,7 @@
 #include "shared/source/os_interface/os_interface.h"
 #include "shared/test/common/helpers/default_hw_info.h"
 #include "shared/test/common/libult/linux/drm_mock.h"
+#include "shared/test/common/mocks/linux/mock_drm_allocation.h"
 
 #include "test.h"
 
@@ -115,3 +116,22 @@ TEST(DrmQueryTest, givenIoctlParamWhenParseToStringThenProperStringIsReturned) {
         EXPECT_STREQ(IoctlHelper::getIoctlParamString(ioctlParamCodeString.first).c_str(), ioctlParamCodeString.second);
     }
 }
+
+TEST(DrmQueryTest, WhenCallingQueryPageFaultSupportThenReturnFalse) {
+    auto executionEnvironment = std::make_unique<ExecutionEnvironment>();
+    executionEnvironment->prepareRootDeviceEnvironments(1);
+    DrmMock drm{*executionEnvironment->rootDeviceEnvironments[0]};
+
+    drm.queryPageFaultSupport();
+
+    EXPECT_FALSE(drm.hasPageFaultSupport());
+}
+
+TEST(DrmQueryTest, givenDrmAllocationWhenShouldAllocationFaultIsCalledThenReturnFalse) {
+    auto executionEnvironment = std::make_unique<ExecutionEnvironment>();
+    executionEnvironment->prepareRootDeviceEnvironments(1);
+    DrmMock drm{*executionEnvironment->rootDeviceEnvironments[0]};
+
+    MockDrmAllocation allocation(GraphicsAllocation::AllocationType::BUFFER, MemoryPool::MemoryNull);
+    EXPECT_FALSE(allocation.shouldAllocationPageFault(&drm));
+}