Add cu::Device::getPower

nlesc-recruit · Sep 24, 2024 · fd144e0 · fd144e0
1 parent 6736c37
commit fd144e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -13,6 +13,7 @@ project adheres to [Semantic Versioning](http://semver.org/).
   `cu::DeviceMemory` object
 - Added `cu::DeviceMemory::memset()`
 - Added `cu::Stream::memsetAsync()`
+- Added `cu::Device::getPower()`
 
 ### Changed
 

diff --git a/include/cudawrappers/nvml.hpp b/include/cudawrappers/nvml.hpp
@@ -54,10 +54,16 @@ class Device {
     return clockMhz;
   }
 
+  unsigned int getPower() {
+    unsigned int power;
+    checkNvmlCall(nvmlDeviceGetPowerUsage(device_, &power));
+    return power;
+  }
+
  private:
   nvmlDevice_t device_;
 };
 }  // namespace nvml
 
 #endif  //  __HIP_PLATFORM_AMD__
-#endif  // NVML_H
+#endif  // NVML_H