61 lines
1.7 KiB
Python
Executable File
61 lines
1.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from subprocess import Popen, PIPE
|
|
import sys
|
|
import os
|
|
import json
|
|
|
|
def run_one(type, name, mode, iters, faster):
|
|
args = []
|
|
res_type = type
|
|
if faster:
|
|
if type == 'proto':
|
|
os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'cpp'
|
|
res_type = 'proto_cpp'
|
|
if type == 'pycapnp':
|
|
args += ['-c', 'packed']
|
|
res_type = 'pycapnp_packed'
|
|
|
|
p = Popen(["time", "-p", "python", "runner.py", name, '-s', type, '-i', str(iters)] + args, stdout=PIPE, stderr=PIPE)
|
|
res = p.communicate()[1]
|
|
|
|
data = {}
|
|
res = res.strip()
|
|
|
|
for line in res.split('\n'):
|
|
vals = line.split()
|
|
data[vals[0]] = float(vals[1])
|
|
|
|
data['type'] = res_type
|
|
data['mode'] = mode
|
|
data['name'] = name
|
|
data['iters'] = iters
|
|
|
|
if 'PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION' in os.environ:
|
|
del os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION']
|
|
return data
|
|
|
|
def run_each(name, iters):
|
|
ret = []
|
|
|
|
ret.append(run_one('pycapnp', name, 'object', iters, False))
|
|
ret.append(run_one('proto', name, 'object', iters, False))
|
|
ret.append(run_one('proto', name, 'object', iters, True))
|
|
|
|
ret.append(run_one('pycapnp', name, 'passByBytes', iters, False))
|
|
ret.append(run_one('pycapnp', name, 'passByBytes', iters, True))
|
|
ret.append(run_one('proto', name, 'passByBytes', iters, False))
|
|
ret.append(run_one('proto', name, 'passByBytes', iters, True))
|
|
|
|
return ret
|
|
|
|
def main():
|
|
data = []
|
|
data += run_each('carsales', 2000)
|
|
data += run_each('catrank', 100)
|
|
data += run_each('eval', 10000)
|
|
json.dump(data, sys.stdout, sort_keys=True, indent=4, separators=(',', ': '))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|