3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
28 lines
708 B
Python
28 lines
708 B
Python
def bm_run(N, M):
|
|
try:
|
|
from utime import ticks_us, ticks_diff
|
|
except ImportError:
|
|
import time
|
|
|
|
ticks_us = lambda: int(time.perf_counter() * 1000000)
|
|
ticks_diff = lambda a, b: a - b
|
|
|
|
# Pick sensible parameters given N, M
|
|
cur_nm = (0, 0)
|
|
param = None
|
|
for nm, p in bm_params.items():
|
|
if 10 * nm[0] <= 12 * N and nm[1] <= M and nm > cur_nm:
|
|
cur_nm = nm
|
|
param = p
|
|
if param is None:
|
|
print(-1, -1, "no matching params")
|
|
return
|
|
|
|
# Run and time benchmark
|
|
run, result = bm_setup(param)
|
|
t0 = ticks_us()
|
|
run()
|
|
t1 = ticks_us()
|
|
norm, out = result()
|
|
print(ticks_diff(t1, t0), norm, out)
|