Reduce the runtime and overhead of the PGO benchmark run to make it work with the constrained resources in the AArch64 docker/qemu build.

Author: scoder

Makefile

@@ -57,7 +57,7 @@ wheel_%: dist/$(PACKAGENAME)-$(VERSION).tar.gz
 				$$PYBIN/pip install -U pip setuptools; \
 				make clean; rm -fr $$PROFDIR; \
 				CFLAGS="$$CFLAGS -fprofile-generate -fprofile-dir=$$PROFDIR" $$PYBIN/python setup.py build_ext -i; \
-				$$PYBIN/python lockbench.py flock; \
+				$$PYBIN/python lockbench.py flock quick; \
 				CFLAGS="$$CFLAGS -fprofile-use -fprofile-correction -fprofile-dir=$$PROFDIR" $$PYBIN/python setup.py build_ext -i -f; \
 				$$PYBIN/python lockbench.py rlock flock; \
 				CFLAGS="$$CFLAGS -fprofile-use -fprofile-correction -fprofile-dir=$$PROFDIR" $$PYBIN/python setup.py bdist_wheel; \

lockbench.py

@@ -98,6 +98,9 @@ def threaded(l, test_func, tcount=10):
     from timeit import Timer
     from functools import partial
 
+    repeat_count = 100000
+    repeat_count_t = 1000
+
     rlock, flock = ('RLock', RLock()), ('FLock', FLock())
     locks = []
     args = sys.argv[1:]
@@ -108,18 +111,19 @@ def threaded(l, test_func, tcount=10):
             locks.append(rlock)
         if 'flock' in args:
             locks.append(flock)
+        for _ in range(args.count('quick')):
+            repeat_count = max(10, repeat_count // 100)
+            repeat_count_t = max(5, repeat_count_t // 10)
     assert locks, args
 
     for name, lock in locks:
         print('Testing %s' % name)
-        repeat_count = 100000
         print("sequential (x%d):" % repeat_count)
         for function in functions:
             timer = Timer(partial(function, lock))
             print('%-25s: %.3f sec' % (function.__name__, max(timer.repeat(repeat=4, number=repeat_count))))
 
-        repeat_count = 1000
-        print("threaded 10T (x%d):" % repeat_count)
+        print("threaded 10T (x%d):" % repeat_count_t)
         for function in functions:
             timer = Timer(partial(threaded, lock, function))
-            print('%-25s: %.3f sec' % (function.__name__, max(timer.repeat(repeat=4, number=repeat_count))))
+            print('%-25s: %.3f sec' % (function.__name__, max(timer.repeat(repeat=4, number=repeat_count_t))))