Endpoints — github.com/Jiayi-Pan/TinyZero

Routeadd

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

tests/ray/test_colocated_workers.py:None

Routecompute_ref_log_prob

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

verl/workers/megatron_workers.py:None

Routecompute_ref_log_prob

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Routecompute_rm_score

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

verl/workers/megatron_workers.py:None

Routecompute_rm_score

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Routecompute_values

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

verl/workers/megatron_workers.py:None

Routecompute_values

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Routedo_nothing

register(dispatch_mode=Dispatch.DP_COMPUTE, blocking=False)

tests/ray/test_data_transfer.py:None

Routeexecute_func_rank_zero

register(dispatch_mode=Dispatch.ALL_TO_ALL, execute_mode=Execute.RANK_ZERO)

verl/single_controller/base/worker.py:None

Routeexecute_with_func_generator

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO_WITH_FUNC)

verl/single_controller/base/worker.py:None

Routefoo

register(dispatch_mode=Dispatch.ONE_TO_ALL, blocking=False)

tests/ray/check_worker_alive/main.py:None

Routefoo_all_to_all

register(Dispatch.ALL_TO_ALL, blocking=False)

tests/ray/test_worker_group_basics.py:None

Routefoo_custom

register(dispatch_mode={'dispatch_fn': two_to_all_dispatch_fn, 'collect_fn': collect_all_to_all})

tests/ray/test_worker_group_basics.py:None

Routefoo_one_to_all

register(Dispatch.ONE_TO_ALL, blocking=False)

tests/ray/test_worker_group_basics.py:None

Routefoo_rank_zero

register(dispatch_mode=Dispatch.ALL_TO_ALL, execute_mode=Execute.RANK_ZERO)

tests/ray/test_worker_group_basics.py:None

Routegenerate_sequences

register(dispatch_mode=Dispatch.MEGATRON_PP_AS_DP_PROTO)

verl/workers/megatron_workers.py:None

Routegenerate_sequences

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Routeinit_model

register(dispatch_mode=Dispatch.ONE_TO_ALL)

tests/ray/detached_worker/server.py:None

Routeinit_model

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/megatron_workers.py:None

Routeinit_model

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/fsdp_workers.py:None

Routeload_checkpoint

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/megatron_workers.py:None

Routeload_pretrained_model

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/megatron_workers.py:None

Routesave_checkpoint

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/megatron_workers.py:None

Routesave_checkpoint

register(dispatch_mode=Dispatch.ONE_TO_ALL)

verl/workers/fsdp_workers.py:None

Routesub

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

tests/ray/test_colocated_workers.py:None

Routetrain_model

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

tests/ray/detached_worker/server.py:None

Routeupdate_actor

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

verl/workers/megatron_workers.py:None

Routeupdate_actor

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Routeupdate_critic

register(dispatch_mode=Dispatch.MEGATRON_COMPUTE_PROTO)

verl/workers/megatron_workers.py:None

Routeupdate_critic

register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)

verl/workers/fsdp_workers.py:None

Endpoints30 in github.com/Jiayi-Pan/TinyZero