From 1be608b81e2dc1b63cb88c3bf8f0cfe10badd67b Mon Sep 17 00:00:00 2001
From: Kamil Kopryk <kamil.kopryk@intel.com>
Date: Fri, 15 Oct 2021 16:04:15 +0000
Subject: [PATCH] Correct unit test

Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
---
 .../unit_test/os_interface/linux/drm_memory_manager_tests.cpp   | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/opencl/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp b/opencl/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp
index 5dd1da2d34..85c228514c 100644
--- a/opencl/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp
+++ b/opencl/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp
@@ -1697,6 +1697,8 @@ HWTEST_F(DrmMemoryManagerTest, givenDrmMemoryManagerWhenTiledImageIsBeingCreated
         GTEST_SKIP();
     }
 
+    device->setPreemptionMode(PreemptionMode::Disabled);
+
     auto csr = static_cast<TestedDrmCommandStreamReceiver<DEFAULT_TEST_FAMILY_NAME> *>(device->getDefaultEngine().commandStreamReceiver);
     csr->callHwFlush = false;