Set CAPTURE flag in fillExecObject

Resolves: NEO-5026

Change-Id: I132d5f1c02edf8e595d31e08029fb32ccbd6d307
Signed-off-by: Mateusz Hoppe <[email protected]>

Author: HoppeMateusz

opencl/test/unit_test/CMakeLists.txt

@@ -190,7 +190,7 @@ function(neo_copy_test_files_with_revision target product revision_id)
                      DEPENDS ${NEO_SOURCE_DIR}/opencl/test/unit_test/test_files
   )
   add_dependencies(${target} copy_compiler_files)
-  set_target_properties(${target} PROPERTIES FOLDER "${PLATFORM_SPECIFIC_TEST_TARGETS_FOLDER}/${product}/${revision_id}}")
+  set_target_properties(${target} PROPERTIES FOLDER "${PLATFORM_SPECIFIC_TEST_TARGETS_FOLDER}/${product}/${revision_id}")
 endfunction()
 
 add_dependencies(unit_tests

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

@@ -430,3 +430,15 @@ TEST(DrmBufferObject, whenMarkForCapturedCalledThenIsMarkedForCaptureReturnsTrue
     bo.markForCapture();
     EXPECT_TRUE(bo.isMarkedForCapture());
 }
+
+TEST_F(DrmBufferObjectTest, givenBoMarkedForCaptureWhenFillingExecObjectThenCaptureFlagIsSet) {
+    drm_i915_gem_exec_object2 execObject;
+
+    memset(&execObject, 0, sizeof(execObject));
+    bo->markForCapture();
+    bo->setAddress(0x45000);
+    bo->setSize(0x1000);
+    bo->fillExecObject(execObject, osContext.get(), 0, 1);
+
+    EXPECT_TRUE(execObject.flags & EXEC_OBJECT_CAPTURE);
+}

shared/source/os_interface/linux/drm_buffer_object.cpp

@@ -115,6 +115,9 @@ void BufferObject::fillExecObject(drm_i915_gem_exec_object2 &execObject, OsConte
     execObject.alignment = 0;
     execObject.offset = this->gpuAddress;
     execObject.flags = EXEC_OBJECT_PINNED | EXEC_OBJECT_SUPPORTS_48B_ADDRESS;
+    if (this->isMarkedForCapture()) {
+        execObject.flags |= EXEC_OBJECT_CAPTURE;
+    }
     execObject.rsvd1 = drmContextId;
     execObject.rsvd2 = 0;