diff --git a/src/main.rs b/src/main.rs index 0f03355..811db88 100644 --- a/src/main.rs +++ b/src/main.rs @@ -378,12 +378,11 @@ fn are_gpus_healthy( tolerate_software_throttling: bool, ) -> (bool, Vec) { let mut reasons = vec![]; - let mut max_flops: f64 = 0.; + // acceptable_flops is tflops_tolerance% lower than best gpu avg flops + let mut acceptable_flops: f64 = 0.; for r in burn_results.iter() { - max_flops = max_flops.max(r.flops_avg()); + acceptable_flops = acceptable_flops.max(r.flops_avg() * (100. - tflops_tolerance) / 100.); } - // acceptable_flops is tflops_tolerance % lower than best gpu avg flops - let acceptable_flops = max_flops * (100. - tflops_tolerance) / 100.; for r in burn_results.iter() { let mut low_flops = false; if r.flops_avg() < acceptable_flops {