NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-30 11:41:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix throttle setters

Browse Source
This commit is contained in:
Yunsong Wang
2025-04-23 14:01:56 -07:00
parent 89bec09b82
commit 31efce1ec8
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nvbench/benchmark_base.cuh
View File

@@ -249,9 +249,10 @@ struct benchmark_base
[[nodiscard]] nvbench::float32_t get_throttle_threshold() const { return m_throttle_threshold; } [[nodiscard]] nvbench::float32_t get_throttle_threshold() const { return m_throttle_threshold; }
void set_throttle_threshold(nvbench::float32_t throttle_threshold) benchmark_base &set_throttle_threshold(nvbench::float32_t throttle_threshold)
{ {
m_throttle_threshold = throttle_threshold; m_throttle_threshold = throttle_threshold;
return *this;
} }
[[nodiscard]] nvbench::float32_t get_throttle_recovery_delay() const [[nodiscard]] nvbench::float32_t get_throttle_recovery_delay() const
@@ -259,9 +260,10 @@ struct benchmark_base
return m_throttle_recovery_delay; return m_throttle_recovery_delay;
} }
void set_throttle_recovery_delay(nvbench::float32_t throttle_recovery_delay) benchmark_base &set_throttle_recovery_delay(nvbench::float32_t throttle_recovery_delay)
{ {
m_throttle_recovery_delay = throttle_recovery_delay; m_throttle_recovery_delay = throttle_recovery_delay;
return *this;
} }
[[nodiscard]] nvbench::criterion_params &get_criterion_params() { return m_criterion_params; } [[nodiscard]] nvbench::criterion_params &get_criterion_params() { return m_criterion_params; }