File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 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)
2828def sdpa (tripy_dtype , torch_dtype ):
Original file line number Diff line number Diff 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
161161def test_tripy_param_update (benchmark ):
You can’t perform that action at this time.
0 commit comments