mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
Add benchmark using queue (#2431)
This commit is contained in:
parent
8e75d150f7
commit
4225ac5081
1 changed files with 31 additions and 0 deletions
31
python/benchmarks/benchmark_queue.py
Normal file
31
python/benchmarks/benchmark_queue.py
Normal 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()
|
Loading…
Add table
Reference in a new issue