Updating Memory API implementations

 - updating zet sysman for memory
 - updating memoryhandlecontext init and memory get by checking
isLocalMemorySupported flag

Change-Id: I084068eb0865bb037b7d80246c4a9c74d3ff2cc1
Signed-off-by: SaiKishore Konda <saikishore.konda@intel.com>

Author: saik-intel

level_zero/api/tools/zet_sysman.cpp

@@ -377,21 +377,21 @@ __zedllexport ze_result_t __zecall
 zetSysmanMemoryGetProperties(
     zet_sysman_mem_handle_t hMemory,
     zet_mem_properties_t *pProperties) {
-    return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    return L0::Memory::fromHandle(hMemory)->memoryGetProperties(pProperties);
 }
 
 __zedllexport ze_result_t __zecall
 zetSysmanMemoryGetState(
     zet_sysman_mem_handle_t hMemory,
     zet_mem_state_t *pState) {
-    return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    return L0::Memory::fromHandle(hMemory)->memoryGetState(pState);
 }
 
 __zedllexport ze_result_t __zecall
 zetSysmanMemoryGetBandwidth(
     zet_sysman_mem_handle_t hMemory,
     zet_mem_bandwidth_t *pBandwidth) {
-    return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+    return L0::Memory::fromHandle(hMemory)->memoryGetBandwidth(pBandwidth);
 }
 
 __zedllexport ze_result_t __zecall

level_zero/tools/source/sysman/memory/CMakeLists.txt

@@ -5,10 +5,10 @@
 #
 
 set(L0_SRCS_TOOLS_SYSMAN_MEMORY
-    ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/memory.cpp
+    ${CMAKE_CURRENT_SOURCE_DIR}/memory.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/memory.h
     ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/memory_imp.cpp
-    ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/memory_imp.h
+    ${CMAKE_CURRENT_SOURCE_DIR}/memory_imp.h
     ${CMAKE_CURRENT_SOURCE_DIR}/os_memory.h
 )
 

level_zero/tools/source/sysman/memory/memory.cpp

@@ -5,8 +5,9 @@
  *
  */
 
-#include "level_zero/tools/source/sysman/memory/memory.h"
+#include "shared/source/memory_manager/memory_manager.h"
 
+#include "level_zero/core/source/device/device.h"
 #include "level_zero/tools/source/sysman/memory/memory_imp.h"
 
 namespace L0 {
@@ -18,13 +19,30 @@ MemoryHandleContext::~MemoryHandleContext() {
 }
 
 ze_result_t MemoryHandleContext::init() {
-    Memory *pMemory = new MemoryImp(pOsSysman, hCoreDevice);
-    handleList.push_back(pMemory);
+    Device *device = L0::Device::fromHandle(hCoreDevice);
+
+    isLmemSupported = device->getDriverHandle()->getMemoryManager()->isLocalMemorySupported(device->getRootDeviceIndex());
+
+    if (isLmemSupported) {
+        Memory *pMemory = new MemoryImp(pOsSysman, hCoreDevice);
+        handleList.push_back(pMemory);
+    }
     return ZE_RESULT_SUCCESS;
 }
 
 ze_result_t MemoryHandleContext::memoryGet(uint32_t *pCount, zet_sysman_mem_handle_t *phMemory) {
-    *pCount = 0;
+    if (nullptr == phMemory) {
+        *pCount = static_cast<uint32_t>(handleList.size());
+        return ZE_RESULT_SUCCESS;
+    }
+    uint32_t i = 0;
+    for (Memory *mem : handleList) {
+        if (i >= *pCount) {
+            break;
+        }
+        phMemory[i++] = mem->toHandle();
+    }
+    *pCount = i;
     return ZE_RESULT_SUCCESS;
 }
 

level_zero/tools/source/sysman/memory/memory.h

@@ -39,6 +39,7 @@ struct MemoryHandleContext {
     ze_result_t memoryGet(uint32_t *pCount, zet_sysman_mem_handle_t *phMemory);
 
     OsSysman *pOsSysman;
+    bool isLmemSupported;
     std::vector<Memory *> handleList;
     ze_device_handle_t hCoreDevice;
 };