diff --git a/src/enzyme_ad/jax/Passes/EqualitySaturation.cpp b/src/enzyme_ad/jax/Passes/EqualitySaturation.cpp index be697b4c1..7513b8e7a 100644 --- 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