Skip to content

Commit 34d2dc1

Browse files
committed
try median?
1 parent 4ca6c53 commit 34d2dc1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/enzyme_ad/jax/Passes/EqualitySaturation.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,7 +483,11 @@ class OperationTimer {
483483
// TODO: This means there's no point in warmup anymore, since we're now
484484
// taking individual measurements. Maybe we get rid of the parameter or do
485485
// something more sophisticated
486-
auto cost = *std::min_element(durations.begin(), durations.end());
486+
//
487+
// auto cost = *std::min_element(durations.begin(), durations.end());
488+
489+
// Try median?
490+
auto cost = durations[durations.size() / 2];
487491
assert(!futures);
488492

489493
// Cleanup

0 commit comments

Comments
 (0)