mirror of
https://github.com/wassname/ray.git
synced 2026-07-18 12:40:56 +08:00
Add benchmark using queue (#2431)
This commit is contained in:
committed by
Philipp Moritz
parent
8e75d150f7
commit
4225ac5081
@@ -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()
|
||||
Reference in New Issue
Block a user