Merge pull request #451 from OctopusET/tenstorrent

Add support for Tenstorrent accelerators

Author: Syllo

CMakeLists.txt

@@ -128,6 +128,13 @@ else()
   set(ROCKCHIP_SUPPORT_DEFAULT OFF)
 endif()
 
+# Tenstorrent support is only available on Linux
+if (CMAKE_SYSTEM_NAME STREQUAL "Linux")
+  set(TENSTORRENT_SUPPORT_DEFAULT ON)
+else()
+  set(TENSTORRENT_SUPPORT_DEFAULT OFF)
+endif()
+
 option(NVIDIA_SUPPORT "Build support for NVIDIA GPUs through libnvml" ${NVIDIA_SUPPORT_DEFAULT})
 option(AMDGPU_SUPPORT "Build support for AMD GPUs through amdgpu driver" ${AMDGPU_SUPPORT_DEFAULT})
 option(RADEON_SUPPORT "Build support for ATI/AMD GPUs through radeon driver" ${RADEON_SUPPORT_DEFAULT})
@@ -142,6 +149,7 @@ option(TPU_SUPPORT "Build support for Google TPUs through GRPC" ${TPU_SUPPORT_DE
 option(ROCKCHIP_SUPPORT "Enable support for Rockchip NPU" ${ROCKCHIP_SUPPORT_DEFAULT})
 option(METAX_SUPPORT "Build support for MetaX GPUs through libmxsml" ${METAX_SUPPORT_DEFAULT})
 option(ENFLAME_SUPPORT "Build support for Enflame GCUs through libefml" ${ENFLAME_SUPPORT_DEFAULT})
+option(TENSTORRENT_SUPPORT "Build support for Tenstorrent accelerators through tt-kmd" ${TENSTORRENT_SUPPORT_DEFAULT})
 option(IXML_SUPPORT "Build support for Iluvatar CoreX GPUs through libixml" ${IXML_SUPPORT_DEFAULT})
 
 add_subdirectory(src)

README.markdown

@@ -11,8 +11,8 @@ htop-familiar way.
 Currently supported vendors are AMD (Linux amdgpu driver), Apple (limited M1 &
 M2 support), Huawei (Ascend), Intel (Linux i915/Xe drivers), NVIDIA (Linux
 proprietary divers), Qualcomm Adreno (Linux MSM driver), Broadcom VideoCore (Linux v3d driver),
-Rockchip, MetaX (MXSML driver), Enflame (Linux EFML driver), Iluvatar CoreX
-(ixML / libixml).
+Rockchip, MetaX (MXSML driver), Enflame (Linux EFML driver), Tenstorrent (Linux tt-kmd driver).
+Rockchip, MetaX (MXSML driver), Enflame (Linux EFML driver), Iluvatar CoreX (ixML / libixml).
 
 Because a picture is worth a thousand words:
 
@@ -37,6 +37,7 @@ Table of Contents
   - [Rockchip](#rockchip)
   - [MetaX](#metax)
   - [Enflame](#enflame)
+  - [Tenstorrent](#tenstorrent)
 - [Build](#build)
 - [Distribution Specific Installation Process](#distribution-specific-installation-process)
   - [Ubuntu / Debian](#ubuntu--debian)
@@ -173,6 +174,12 @@ NVTOP supports Enflame GCUs (testing on Enflame S60, Enflame L300 and Enflame L6
 
 GCU, which refers to General Compute Unit, is a type of accelerator card that is used to perform general-purpose computing tasks just like GPGPU.
 
+### Tenstorrent
+
+NVTOP supports Tenstorrent AI accelerators (Blackhole, Wormhole, Grayskull) through the [tt-kmd](https://github.com/tenstorrent/tt-kmd) kernel driver.
+
+Supports temperature, power draw, AI clock, fan RPM, PCIe link info, and process listing. No external libraries required -- all data is read from sysfs, hwmon, and procfs.
+
 Build
 -----
 

src/CMakeLists.txt

@@ -164,6 +164,10 @@ if(ENFLAME_SUPPORT)
   target_sources(nvtop PRIVATE extract_gcuinfo_enflame.c)
 endif()
 
+if(TENSTORRENT_SUPPORT)
+  target_sources(nvtop PRIVATE extract_gpuinfo_tenstorrent.c)
+endif()
+
 if(IXML_SUPPORT)
   message(STATUS "Building with Iluvatar CoreX GPU support")
   target_sources(nvtop PRIVATE extract_gpuinfo_ixml.c)