From 12de668ccbe1dc797edc678868fa3ee26a762c78 Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Thu, 31 May 2018 00:06:17 -0700 Subject: [PATCH] [ASV] Add ray.init and simple Ray benchmarks (#2166) --- python/benchmarks/benchmarks.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/python/benchmarks/benchmarks.py b/python/benchmarks/benchmarks.py index e374d19a2..b5b6202ad 100644 --- a/python/benchmarks/benchmarks.py +++ b/python/benchmarks/benchmarks.py @@ -2,6 +2,13 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import ray +import ray.test.test_functions as test_functions + + +def setup(): + ray.init(num_workers=4, num_cpus=4) + class TimeSuite(object): """An example benchmark.""" @@ -24,3 +31,15 @@ class TimeSuite(object): class MemSuite(object): def mem_list(self): return [0] * 256 + + +class MicroBenchmarkSuite(object): + def time_submit(self): + test_functions.empty_function.remote() + + def time_submit_and_get(self): + x = test_functions.trivial_function.remote() + ray.get(x) + + def time_put(self): + ray.put(1)