Add benchmark using queue (#2431)

This commit is contained in:
Peter Schafhalter 2018-07-19 16:43:22 -07:00 committed by Philipp Moritz
parent 8e75d150f7
commit 4225ac5081

View file

@ -0,0 +1,31 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import ray
from ray.experimental.queue import Queue
def setup():
if not hasattr(setup, "is_initialized"):
ray.init(num_workers=4, num_cpus=4)
setup.is_initialized = True
class QueueSuite(object):
def time_put(self):
queue = Queue(1000)
for i in range(1000):
queue.put(i)
def time_get(self):
queue = Queue()
for i in range(1000):
queue.put(i)
for _ in range(1000):
queue.get()
def time_qsize(self):
queue = Queue()
for _ in range(1000):
queue.qsize()