fix: enable eustall only if kmd supports

Related-To: NEO-15193

Signed-off-by: Joshua Santosh Ranjan <joshua.santosh.ranjan@intel.com>
Source: 5b67a3d

Author: joshuaranjan

level_zero/tools/source/metrics/metric.cpp

@@ -121,6 +121,7 @@ ze_result_t MetricDeviceContext::metricGroupGet(uint32_t *pCount, zet_metric_gro
             }
         }
     }
+
     *pCount = availableCount;
     return result;
 }

level_zero/tools/source/metrics/metric_ip_sampling_source.cpp

@@ -56,7 +56,7 @@ bool IpSamplingMetricSourceImp::isAvailable() {
     return isEnabled;
 }
 
-void IpSamplingMetricSourceImp::cacheMetricGroup() {
+ze_result_t IpSamplingMetricSourceImp::cacheMetricGroup() {
 
     const auto deviceImp = static_cast<DeviceImp *>(&metricDeviceContext.getDevice());
     if (metricDeviceContext.isImplicitScalingCapable()) {
@@ -70,14 +70,26 @@ void IpSamplingMetricSourceImp::cacheMetricGroup() {
             uint32_t count = 1;
             zet_metric_group_handle_t hMetricGroup = {};
             const auto result = source.metricGroupGet(&count, &hMetricGroup);
+            if (result == ZE_RESULT_ERROR_UNSUPPORTED_FEATURE) {
+                return result;
+            }
             // Getting MetricGroup from sub-device cannot fail, since RootDevice is successful
             UNRECOVERABLE_IF(result != ZE_RESULT_SUCCESS);
             subDeviceMetricGroup.push_back(static_cast<IpSamplingMetricGroupImp *>(MetricGroup::fromHandle(hMetricGroup)));
         }
 
         IpSamplingMetricSourceImp &source = deviceImp->getMetricDeviceContext().getMetricSource<IpSamplingMetricSourceImp>();
         cachedMetricGroup = MultiDeviceIpSamplingMetricGroupImp::create(source, subDeviceMetricGroup);
-        return;
+        return ZE_RESULT_SUCCESS;
+    }
+
+    // Confirm whether sample collection is possible
+    uint32_t referenceValues = 100;
+    const ze_result_t sampleCheckResult = getMetricOsInterface()->startMeasurement(referenceValues, referenceValues);
+    if (sampleCheckResult != ZE_RESULT_SUCCESS) {
+        return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    } else {
+        getMetricOsInterface()->stopMeasurement();
     }
 
     std::vector<IpSamplingMetricImp> metrics = {};
@@ -114,6 +126,8 @@ void IpSamplingMetricSourceImp::cacheMetricGroup() {
 
     cachedMetricGroup = IpSamplingMetricGroupImp::create(*this, metrics);
     DEBUG_BREAK_IF(cachedMetricGroup == nullptr);
+
+    return ZE_RESULT_SUCCESS;
 }
 
 ze_result_t IpSamplingMetricSourceImp::metricGroupGet(uint32_t *pCount, zet_metric_group_handle_t *phMetricGroups) {
@@ -123,15 +137,21 @@ ze_result_t IpSamplingMetricSourceImp::metricGroupGet(uint32_t *pCount, zet_metr
         return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
     }
 
+    if (cachedMetricGroup == nullptr) {
+        auto status = cacheMetricGroup();
+
+        if (status != ZE_RESULT_SUCCESS) {
+            *pCount = 0;
+            isEnabled = false;
+            return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+        }
+    }
+
     if (*pCount == 0) {
         *pCount = 1;
         return ZE_RESULT_SUCCESS;
     }
 
-    if (cachedMetricGroup == nullptr) {
-        cacheMetricGroup();
-    }
-
     DEBUG_BREAK_IF(phMetricGroups == nullptr);
     phMetricGroups[0] = cachedMetricGroup->toHandle();
     *pCount = 1;

level_zero/tools/source/metrics/metric_ip_sampling_source.h

@@ -67,7 +67,7 @@ class IpSamplingMetricSourceImp : public MetricSource {
     bool canDisable() override;
 
   protected:
-    void cacheMetricGroup();
+    ze_result_t cacheMetricGroup();
     bool isEnabled = false;
 
     const MetricDeviceContext &metricDeviceContext;

level_zero/tools/test/unit_tests/sources/metrics/test_metric_ip_sampling_enumeration.cpp

@@ -90,6 +90,22 @@ HWTEST2_F(MetricIpSamplingEnumerationTest, GivenDependenciesAvailableWhenMetricG
     }
 }
 
+HWTEST2_F(MetricIpSamplingEnumerationTest, GivenDependenciesNotAvailableWhenMetricGroupGetIsCalledThenValidMetricGroupIsReturned, EustallSupportedPlatforms) {
+
+    EXPECT_EQ(ZE_RESULT_SUCCESS, testDevices[0]->getMetricDeviceContext().enableMetricApi());
+
+    for (auto &osInterface : osInterfaceVector) {
+        osInterface->startMeasurementReturn = ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    }
+
+    for (auto device : testDevices) {
+
+        uint32_t metricGroupCount = 0;
+        EXPECT_EQ(zetMetricGroupGet(device->toHandle(), &metricGroupCount, nullptr), ZE_RESULT_SUCCESS);
+        EXPECT_EQ(metricGroupCount, 0u);
+    }
+}
+
 HWTEST2_F(MetricIpSamplingEnumerationTest, GivenDependenciesAvailableWhenMetricGroupGetIsCalledMultipleTimesThenValidMetricGroupIsReturned, EustallSupportedPlatforms) {
 
     EXPECT_EQ(ZE_RESULT_SUCCESS, testDevices[0]->getMetricDeviceContext().enableMetricApi());

level_zero/tools/test/unit_tests/sources/metrics/test_metric_ip_sampling_streamer.cpp

@@ -133,8 +133,6 @@ TEST_F(MetricIpSamplingStreamerTest, GivenStartMeasurementFailsWhenStreamerOpenI
             subDeviceIndex = 2;
         }
 
-        osInterfaceVector[subDeviceIndex]->startMeasurementReturn = ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
-
         zet_metric_group_handle_t metricGroupHandle = MetricIpSamplingStreamerTest::getMetricGroup(device);
         EXPECT_EQ(zetContextActivateMetricGroups(context->toHandle(), device, 1, &metricGroupHandle), ZE_RESULT_SUCCESS);
 
@@ -145,6 +143,7 @@ TEST_F(MetricIpSamplingStreamerTest, GivenStartMeasurementFailsWhenStreamerOpenI
         streamerDesc.notifyEveryNReports = 32768;
         streamerDesc.samplingPeriod = 1000;
 
+        osInterfaceVector[subDeviceIndex]->startMeasurementReturn = ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
         EXPECT_EQ(
             zetMetricStreamerOpen(context->toHandle(), device, metricGroupHandle, &streamerDesc, eventHandle, &streamerHandle),
             osInterfaceVector[subDeviceIndex]->startMeasurementReturn);