Fix throttle setters

PointKernel · PointKernel · commit 31efce1ec828 · 2025-04-23T14:01:56.000-07:00
diff --git a/nvbench/benchmark_base.cuh b/nvbench/benchmark_base.cuh
@@ -249,19 +249,21 @@ 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
   {
     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; }

Original file line number	Diff line number	Diff line change
`@@ -249,19 +249,21 @@ struct benchmark_base`
`249`	`249`
`250`	`250`	`[[nodiscard]] nvbench::float32_t get_throttle_threshold() const { return m_throttle_threshold; }`
`251`	`251`
`252`		`- void set_throttle_threshold(nvbench::float32_t throttle_threshold)`
	`252`	`+ benchmark_base &set_throttle_threshold(nvbench::float32_t throttle_threshold)`
`253`	`253`	`{`
`254`	`254`	`m_throttle_threshold = throttle_threshold;`
	`255`	`+ return *this;`
`255`	`256`	`}`
`256`	`257`
`257`	`258`	`[[nodiscard]] nvbench::float32_t get_throttle_recovery_delay() const`
`258`	`259`	`{`
`259`	`260`	`return m_throttle_recovery_delay;`
`260`	`261`	`}`
`261`	`262`
`262`		`- void set_throttle_recovery_delay(nvbench::float32_t throttle_recovery_delay)`
	`263`	`+ benchmark_base &set_throttle_recovery_delay(nvbench::float32_t throttle_recovery_delay)`
`263`	`264`	`{`
`264`	`265`	`m_throttle_recovery_delay = throttle_recovery_delay;`
	`266`	`+ return *this;`
`265`	`267`	`}`
`266`	`268`
`267`	`269`	`[[nodiscard]] nvbench::criterion_params &get_criterion_params() { return m_criterion_params; }`