Skip to content

vllm.ray.lazy_utils

Functions:

is_in_ray_actor()

Check if we are in a Ray actor.

Source code in vllm/ray/lazy_utils.py 
17
18
19
20
21
22
23
24
25
26
27
28
29
30
def is_in_ray_actor():
    """Check if we are in a Ray actor."""

    try:
        import ray

        return (
            ray.is_initialized()
            and ray.get_runtime_context().get_actor_id() is not None
        )
    except ImportError:
        return False
    except AttributeError:
        return False

is_ray_initialized()

Check if Ray is initialized.

Source code in vllm/ray/lazy_utils.py 
 5
 6
 7
 8
 9
10
11
12
13
14
def is_ray_initialized():
    """Check if Ray is initialized."""
    try:
        import ray

        return ray.is_initialized()
    except ImportError:
        return False
    except AttributeError:
        return False