Skip to content

Commit ac307a5

Browse files
Relaxes some perf tolerances
1 parent 1583d86 commit ac307a5

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tripy/tests/performance/cases/sdpa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
@perf_fixture(
2323
params=[
2424
PerfParam(tp.float32, perf_threshold=1.20),
25-
PerfParam(tp.float16, perf_threshold=1.32),
25+
PerfParam(tp.float16, perf_threshold=1.28),
2626
]
2727
)
2828
def sdpa(tripy_dtype, torch_dtype):

tripy/tests/performance/test_perf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,9 @@ def measure_thunk():
153153
print(f"deltas: {deltas}")
154154
assert all(delta < 45 for delta in deltas)
155155

156-
# Ensure all deltas are within a few microseconds of each other
156+
# Ensure all deltas are within a few microseconds of the average
157157
average_delta = sum(deltas) / float(len(deltas))
158-
assert all(abs(delta - average_delta) < 10 for delta in deltas)
158+
assert all(delta - average_delta < 10 for delta in deltas)
159159

160160

161161
def test_tripy_param_update(benchmark):

0 commit comments

Comments
 (0)