diff --git a/libhsakmt.h b/libhsakmt.h index 0f852ead65..031ceb33cf 100644 --- a/libhsakmt.h +++ b/libhsakmt.h @@ -36,6 +36,7 @@ #include "impl/wddm/device.h" wsl::thunk::WDDMDevice* get_wddmdev(uint32_t node_id); +uint32_t get_num_wddmdev(); wsl::thunk::GpuMemory *get_gpu_mem(void *MemoryAddress); #define HSAKMT_DEBUG_LEVEL_ERR -1 diff --git a/topology.cpp b/topology.cpp index e9e34be8d6..397477e534 100644 --- a/topology.cpp +++ b/topology.cpp @@ -1461,3 +1461,7 @@ wsl::thunk::WDDMDevice *get_wddmdev(uint32_t node_id) { return dxg_topology->wdevices_[node_id - 1]; } + +uint32_t get_num_wddmdev() { + return dxg_topology->wdevices_.size(); +}