only report HBM free memory

Signed-off-by: Sergey V Maslov <[email protected]>

Author: smaslov-intel

sycl/plugins/level_zero/pi_level_zero.cpp

@@ -3144,6 +3144,12 @@ pi_result piDeviceGetInfo(pi_device Device, pi_device_info ParamName,
             (ZeDevice, &MemCount, MemHandles.data()));
 
     for (auto &ZesMemHandle : MemHandles) {
+      ZesStruct<zes_mem_properties_t> ZeMemProperties;
+      ZE_CALL(zesMemoryGetProperties, (ZesMemHandle, &ZeMemProperties));
+      // Only report HBM which zeMemAllocDevice allocates from.
+      if (ZeMemProperties.type != ZES_MEM_TYPE_HBM)
+        continue;
+
       ZesStruct<zes_mem_state_t> ZeMemState;
       ZE_CALL(zesMemoryGetState, (ZesMemHandle, &ZeMemState));
       FreeMemory += ZeMemState.free;

sycl/plugins/level_zero/pi_level_zero.hpp

@@ -170,6 +170,10 @@ template <> zes_structure_type_t getZesStructureType<zes_mem_state_t>() {
   return ZES_STRUCTURE_TYPE_MEM_STATE;
 }
 
+template <> zes_structure_type_t getZesStructureType<zes_mem_properties_t>() {
+  return ZES_STRUCTURE_TYPE_MEM_PROPERTIES;
+}
+
 // The helpers to properly default initialize Level-Zero descriptor and
 // properties structures.
 template <class T> struct ZeStruct : public T {