test-ray.py

import time
import ray

delay = 0.01

context = ray.init()
num_cpus = ray.available_resources()["CPU"]


def busy_wait(duration):
    time_perf_counter = time.perf_counter
    start_time = time_perf_counter()
    while (time_perf_counter() - start_time) < duration:
        pass


@ray.remote
def f():
    busy_wait(delay)
    return 1


@ray.remote
def g():
    return sum(ray.get([f.remote() for _ in range(200)]))


print(sum(ray.get([g.remote() for _ in range(500)])) * delay / num_cpus)