mirror of
https://github.com/vale981/ray
synced 2025-03-09 21:06:39 -04:00
32 lines
702 B
Python
32 lines
702 B
Python
|
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()
|