Add support for Tenstorrent accelerators

CMakeLists.txt

@@ -122,6 +122,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(INTEL_SUPPORT "Build support for Intel GPUs through i915 or xe driver" ${INTEL_SUPPORT_DEFAULT})
@@ -135,6 +142,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})
 
 add_subdirectory(src)
 

README.markdown

@@ -11,7 +11,7 @@ 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).
+Rockchip, MetaX (MXSML driver), Enflame (Linux EFML driver), Tenstorrent (Linux tt-kmd driver).
 
 Because a picture is worth a thousand words:
 
@@ -35,6 +35,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)
@@ -160,6 +161,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

@@ -160,6 +160,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()
+
 target_include_directories(nvtop PRIVATE
   ${PROJECT_SOURCE_DIR}/include
   ${PROJECT_BINARY_DIR}/include)