fix: Update Sysman VF management CTS (#142)

fix: Update Sysman VF management CTS

Related-To: VLCLJ-2390

Signed-off-by: B, Vishnu Khanth <[email protected]>

Author: vishnu-khanth

conformance_tests/sysman/test_sysman_vf_management/src/test_sysman_vf_management.cpp

@@ -26,7 +26,7 @@ class VfManagementTest : public lzt::SysmanCtsClass {
 #define VF_MANAGEMENT_TEST VfManagementTest
 #endif // USE_ZESINIT
 
-void validate_vf_capabilities(zes_vf_exp_capabilities_t &vf_capabilities,
+void validate_vf_capabilities(zes_vf_exp2_capabilities_t &vf_capabilities,
                               uint32_t &vf_count) {
   EXPECT_GE(vf_capabilities.address.domain, 0);
   EXPECT_LE(vf_capabilities.address.domain, MAX_DOMAINs);
@@ -41,8 +41,8 @@ void validate_vf_capabilities(zes_vf_exp_capabilities_t &vf_capabilities,
   EXPECT_LE(vf_capabilities.vfID, vf_count);
 }
 
-void compare_vf_capabilities(zes_vf_exp_capabilities_t &vf_capability_initial,
-                             zes_vf_exp_capabilities_t &vf_capability_later) {
+void compare_vf_capabilities(zes_vf_exp2_capabilities_t &vf_capability_initial,
+                             zes_vf_exp2_capabilities_t &vf_capability_later) {
   EXPECT_EQ(vf_capability_initial.stype, vf_capability_later.stype);
   EXPECT_EQ(vf_capability_initial.pNext, vf_capability_later.pNext);
   EXPECT_EQ(vf_capability_initial.address.domain,

utils/test_harness/sysman/include/test_harness_sysman_vf_management.hpp

@@ -16,7 +16,7 @@ namespace level_zero_tests {
 uint32_t get_vf_handles_count(zes_device_handle_t device);
 std::vector<zes_vf_handle_t> get_vf_handles(zes_device_handle_t device,
                                             uint32_t &count);
-zes_vf_exp_capabilities_t get_vf_capabilities(zes_vf_handle_t vf_handle);
+zes_vf_exp2_capabilities_t get_vf_capabilities(zes_vf_handle_t vf_handle);
 uint32_t get_vf_mem_util_count(zes_vf_handle_t vf_handle);
 std::vector<zes_vf_util_mem_exp2_t> get_vf_mem_util(zes_vf_handle_t vf_handle,
                                                     uint32_t &count);

utils/test_harness/sysman/src/test_harness_sysman_vf_management.cpp

@@ -31,10 +31,11 @@ std::vector<zes_vf_handle_t> get_vf_handles(zes_device_handle_t device,
   return vf_handles;
 }
 
-zes_vf_exp_capabilities_t get_vf_capabilities(zes_vf_handle_t vf_handle) {
-  zes_vf_exp_capabilities_t vf_capabilities{};
+zes_vf_exp2_capabilities_t get_vf_capabilities(zes_vf_handle_t vf_handle) {
+  zes_vf_exp2_capabilities_t vf_capabilities = {
+      ZES_STRUCTURE_TYPE_VF_EXP2_CAPABILITIES};
   EXPECT_EQ(ZE_RESULT_SUCCESS,
-            zesVFManagementGetVFCapabilitiesExp(vf_handle, &vf_capabilities));
+            zesVFManagementGetVFCapabilitiesExp2(vf_handle, &vf_capabilities));
   return vf_capabilities;
 }
 
@@ -51,6 +52,9 @@ std::vector<zes_vf_util_mem_exp2_t> get_vf_mem_util(zes_vf_handle_t vf_handle,
     count = get_vf_mem_util_count(vf_handle);
   }
   std::vector<zes_vf_util_mem_exp2_t> vf_util_mem_exp(count);
+  for (uint32_t i = 0; i < count; i++) {
+    vf_util_mem_exp[i].stype = ZES_STRUCTURE_TYPE_VF_UTIL_MEM_EXP2;
+  }
   EXPECT_EQ(ZE_RESULT_SUCCESS, zesVFManagementGetVFMemoryUtilizationExp2(
                                    vf_handle, &count, vf_util_mem_exp.data()));
   return vf_util_mem_exp;
@@ -69,6 +73,9 @@ get_vf_engine_util(zes_vf_handle_t vf_handle, uint32_t &count) {
     count = get_vf_engine_util_count(vf_handle);
   }
   std::vector<zes_vf_util_engine_exp2_t> vf_util_engine_exp(count);
+  for (uint32_t i = 0; i < count; i++) {
+    vf_util_engine_exp[i].stype = ZES_STRUCTURE_TYPE_VF_UTIL_ENGINE_EXP2;
+  }
   EXPECT_EQ(ZE_RESULT_SUCCESS,
             zesVFManagementGetVFEngineUtilizationExp2(
                 vf_handle, &count, vf_util_engine_exp.data()));