feature: query wmtp capability from xe engine

Related-To: NEO-10445
Signed-off-by: Mateusz Jablonski <[email protected]>

Author: JablonskiMateusz

shared/source/os_interface/linux/engine_info.cpp

@@ -84,6 +84,10 @@ void EngineInfo::mapEngine(const NEO::IoctlHelper *ioctlHelper, const EngineCapa
         engineCounters.numComputeEngines++;
     }
     if (engineType != aub_stream::EngineType::NUM_ENGINES) {
+        if (engineInfo.capabilities.wmtpSupport) {
+            auto &wmtpInfoMask = rootDeviceEnvironment.getMutableHardwareInfo()->featureTable.wmtpInfoMask;
+            wmtpInfoMask.set(static_cast<uint32_t>(engineType));
+        }
         tileToEngineToInstanceMap[tileId][engineType] = engine;
     }
 }

shared/source/os_interface/linux/ioctl_helper.h

@@ -51,6 +51,7 @@ struct EngineCapabilities {
     struct Flags {
         bool copyClassSaturatePCIE;
         bool copyClassSaturateLink;
+        bool wmtpSupport;
     };
     Flags capabilities;
 };

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

@@ -254,7 +254,9 @@ std::unique_ptr<EngineInfo> IoctlHelperXe::createEngineInfo(bool isSysmanEnabled
             if (enginesPerTile.size() <= tile) {
                 enginesPerTile.resize(tile + 1);
             }
-            enginesPerTile[tile].push_back({engineClassInstance, {}});
+            EngineCapabilities::Flags engineFlags{};
+            engineFlags.wmtpSupport = isValueSet(queryEngines->engines[i].capabilities, DRM_XE_ENGINE_CAPABILITY_WMTP);
+            enginesPerTile[tile].push_back({engineClassInstance, engineFlags});
             if (!defaultEngine && engineClassInstance.engineClass == defaultEngineClass) {
                 defaultEngine = std::make_unique<drm_xe_engine_class_instance>();
                 *defaultEngine = engine;

shared/source/sku_info/sku_info_base.h

@@ -7,13 +7,16 @@
 
 #pragma once
 
+#include "aubstream/engine_node.h"
+
 #include <array>
 #include <bitset>
 #include <cstdint>
 
 namespace NEO {
 constexpr size_t bcsInfoMaskSize = 9u;
 using BcsInfoMask = std::bitset<bcsInfoMaskSize>;
+using WmtpInfoMask = std::bitset<aub_stream::EngineType::NUM_ENGINES>;
 
 struct FeatureTableBase {
   public:
@@ -75,6 +78,7 @@ struct FeatureTableBase {
     };
 
     BcsInfoMask ftrBcsInfo = 1;
+    WmtpInfoMask wmtpInfoMask{};
 
     union {
         Flags flags;

shared/test/unit_test/os_interface/linux/xe/ioctl_helper_xe_tests.cpp

@@ -1177,6 +1177,23 @@ TEST(IoctlHelperXeTest, givenMissingSupportedTopologiesWhenGetTopologyDataAndMap
     EXPECT_FALSE(result);
 }
 
+TEST(IoctlHelperXeTest, whenCreatingEngineInfoThenWmtpInfoMaskIsProperlySet) {
+    DebugManagerStateRestore restorer;
+    auto executionEnvironment = std::make_unique<MockExecutionEnvironment>();
+    auto drm = DrmMockXe::create(*executionEnvironment->rootDeviceEnvironments[0]);
+    auto xeIoctlHelper = static_cast<MockIoctlHelperXe *>(drm->getIoctlHelper());
+    xeIoctlHelper->initialize();
+    auto &hwInfo = *executionEnvironment->rootDeviceEnvironments[0]->getHardwareInfo();
+    EXPECT_EQ(0u, hwInfo.featureTable.wmtpInfoMask.to_ulong());
+    auto engineInfo = xeIoctlHelper->createEngineInfo(false);
+    EXPECT_FALSE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_CCS)));
+    EXPECT_TRUE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_CCS1)));
+    EXPECT_FALSE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_CCS2)));
+    EXPECT_FALSE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_CCS3)));
+    EXPECT_FALSE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_RCS)));
+    EXPECT_FALSE(hwInfo.featureTable.wmtpInfoMask.test(static_cast<uint32_t>(aub_stream::EngineType::ENGINE_CCCS)));
+}
+
 TEST(IoctlHelperXeTest, whenCreatingEngineInfoThenProperEnginesAreDiscovered) {
     DebugManagerStateRestore restorer;
     auto executionEnvironment = std::make_unique<MockExecutionEnvironment>();

shared/test/unit_test/os_interface/linux/xe/ioctl_helper_xe_tests.h

@@ -342,9 +342,9 @@ struct DrmMockXe : public DrmMockCustom {
         xeQueryEngines->engines[1] = {{DRM_XE_ENGINE_CLASS_COPY, 1, 0}, {}};
         xeQueryEngines->engines[2] = {{DRM_XE_ENGINE_CLASS_COPY, 2, 0}, {}};
         xeQueryEngines->engines[3] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 3, 0}, {}};
-        xeQueryEngines->engines[4] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 4, 0}, {}};
+        xeQueryEngines->engines[4] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 4, 0}, DRM_XE_ENGINE_CAPABILITY_WMTP};
         xeQueryEngines->engines[5] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 5, 1}, {}};
-        xeQueryEngines->engines[6] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 6, 1}, {}};
+        xeQueryEngines->engines[6] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 6, 1}, DRM_XE_ENGINE_CAPABILITY_WMTP};
         xeQueryEngines->engines[7] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 7, 1}, {}};
         xeQueryEngines->engines[8] = {{DRM_XE_ENGINE_CLASS_COMPUTE, 8, 1}, {}};
         xeQueryEngines->engines[9] = {{DRM_XE_ENGINE_CLASS_VIDEO_DECODE, 9, 1}, {}};

third_party/uapi/xe/.version

@@ -0,0 +1 @@
+patch: https://patchwork.freedesktop.org/patch/604493/?series=13613

third_party/uapi/xe/xe_drm.h

@@ -243,13 +243,21 @@ struct drm_xe_engine_class_instance {
 
 /**
  * struct drm_xe_engine - describe hardware engine
- */
+ *
+ * The @capabilities can be:
+ *  - DRM_XE_ENGINE_CAPABILITY_WMTP - represents a engine's mid-thread
+ *  preemption capability.
+*/
 struct drm_xe_engine {
 	/** @instance: The @drm_xe_engine_class_instance */
 	struct drm_xe_engine_class_instance instance;
-
+#define DRM_XE_ENGINE_CAPABILITY_WMTP		1
+	/** @capabilities: Capabilities of this engine. */
+	__u32 capabilities;
+	/** @pad: MBZ */
+	__u32 pad;
 	/** @reserved: Reserved */
-	__u64 reserved[3];
+	__u64 reserved[2];
 };
 
 /**