wip

huggingface · Jan 29, 2025 · d9da9bb · d9da9bb
1 parent ed93d3a
commit d9da9bb
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -378,12 +378,11 @@ fn are_gpus_healthy(
     tolerate_software_throttling: bool,
 ) -> (bool, Vec<String>) {
     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 {