feat(common) adding device utilities

2024-09-28 21:10:18 +00:00 · 2024-09-28 21:10:18 +00:00 · fb81ffce02
parent 5f9120da9c
commit fb81ffce02
1 changed files with 1 additions and 1 deletions
--- a/csrc/common/lib/device.cpp
+++ b/csrc/common/lib/device.cpp
@ -1,7 +1,7 @@
 //
 // Created by morgan on 27/09/24.
 //
-
+#include <nvml.h>
 #include "device.hpp"

 std::optional<nvmlDevice_t> huggingface::tgi::GetDeviceByIndex(device_index_t device)