try median?

EnzymeAD · Jan 31, 2025 · 34d2dc1 · 34d2dc1
1 parent 4ca6c53
commit 34d2dc1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/EqualitySaturation.cpp b/src/enzyme_ad/jax/Passes/EqualitySaturation.cpp
@@ -483,7 +483,11 @@ class OperationTimer {
     // TODO: This means there's no point in warmup anymore, since we're now
     // taking individual measurements. Maybe we get rid of the parameter or do
     // something more sophisticated
-    auto cost = *std::min_element(durations.begin(), durations.end());
+    //
+    // auto cost = *std::min_element(durations.begin(), durations.end());
+
+    // Try median?
+    auto cost = durations[durations.size() / 2];
     assert(!futures);
 
     // Cleanup