NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #214 from PointKernel/fix-throttle-setters

Browse Source 
Fix throttle setter return values and update customization example
This commit is contained in:
Allison Piper
2025-04-24 13:53:20 -04:00
committed by GitHub
parent 2938a94d49 dbd12f61b8
commit 0573ffa9bd
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; }
void set_throttle_threshold(nvbench::float32_t throttle_threshold)
benchmark_base &set_throttle_threshold(nvbench::float32_t throttle_threshold)
{
m_throttle_threshold = throttle_threshold;
return *this;
}
[[nodiscard]] nvbench::float32_t get_throttle_recovery_delay() const
@@ -259,9 +260,10 @@ struct benchmark_base
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;
return *this;
}
[[nodiscard]] nvbench::criterion_params &get_criterion_params() { return m_criterion_params; }