Merge pull request #398 from zhenyu-xu-metax/add-feature-support-metax

Syllo · web-flow · commit 79de353cdcf9 · 2025-10-25T14:32:20.000+02:00
Feat: add support for METAX GPU
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -70,6 +70,7 @@ if(APPLE)
   set(PANFROST_SUPPORT_DEFAULT OFF)
   set(PANTHOR_SUPPORT_DEFAULT OFF)
   set(ASCEND_SUPPORT_DEFAULT OFF)
+  set(METAX_SUPPORT_DEFAULT OFF)
 elseif(ASCEND_SUPPORT)
   set(APPLE_SUPPORT_DEFAULT OFF)
   set(NVIDIA_SUPPORT_DEFAULT OFF)
@@ -79,6 +80,7 @@ elseif(ASCEND_SUPPORT)
   set(PANFROST_SUPPORT_DEFAULT OFF)
   set(PANTHOR_SUPPORT_DEFAULT OFF)
   set(ASCEND_SUPPORT_DEFAULT ON)
+  set(METAX_SUPPORT_DEFAULT OFF)
 else()
   set(APPLE_SUPPORT_DEFAULT OFF)
   set(NVIDIA_SUPPORT_DEFAULT ON)
@@ -89,6 +91,7 @@ else()
   set(PANFROST_SUPPORT_DEFAULT ON)
   set(PANTHOR_SUPPORT_DEFAULT ON)
   set(ASCEND_SUPPORT_DEFAULT OFF)
+  set(METAX_SUPPORT_DEFAULT ON)
 endif()
 
 # TPU support is only available on Linux
@@ -128,6 +131,7 @@ option(ASCEND_SUPPORT "Build support for Ascend NPUs through Ascend DCMI" ${ASCE
 option(V3D_SUPPORT "Build support for Raspberrypi through v3d" ${V3D_SUPPORT_DEFAULT})
 option(TPU_SUPPORT "Build support for Google TPUs through GRPC" ${TPU_SUPPORT_DEFAULT})
 option(ROCKCHIP_SUPPORT "Enable support for Rockchip NPU" ${ROCKCHIP_SUPPORT_DEFAULT})
+option(METAX_SUPPORT "Build support for MetaX GPUs through libmxsml" ${METAX_SUPPORT_DEFAULT})
 
 add_subdirectory(src)
 
diff --git a/README.markdown b/README.markdown
@@ -144,6 +144,12 @@ NVTOP supports Rockchip (testing on orangepi 5 plus).
 
 Supports NPU frequency, temperature, utilization.
 
+### MetaX
+
+NVTOP supports MetaX (testing on MXC500) by MXSML LIBRARY.
+
+For more information about GPUs please take a look at the [METAX documentation](https://developer.metax-tech.com/doc/index)
+
 Build
 -----
 
@@ -154,6 +160,8 @@ Several libraries are required in order for NVTOP to display GPU info:
 * For NVIDIA: the *NVIDIA Management Library* (*NVML*) which comes with the GPU driver.
   * This queries the GPU for info.
 * For AMD: the libdrm library used to query AMD GPUs through the kernel driver.
+* For METAX: the *MetaX System Management Library* (*MXSML*) which comes with the GPU driver.
+  * This queries the GPU for info.
 
 ## Distribution Specific Installation Process
 
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -151,6 +151,10 @@ if(ROCKCHIP_SUPPORT)
   target_sources(nvtop PRIVATE extract_npuinfo_rockchip.c)
 endif()
 
+if(METAX_SUPPORT)
+  target_sources(nvtop PRIVATE extract_gpuinfo_metax.c)
+endif()
+
 target_include_directories(nvtop PRIVATE
   ${PROJECT_SOURCE_DIR}/include
   ${PROJECT_BINARY_DIR}/include)
diff --git a/src/extract_gpuinfo_metax.c b/src/extract_gpuinfo_metax.c
