mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21: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