nemo_rl.distributed.worker_group_utils#

Module Contents#

Functions#

get_nsight_config_if_pattern_matches

Check if worker name matches patterns in NRL_NSYS_WORKER_PATTERNS and return nsight config.

recursive_merge_options

Recursively merge extra options into default options using OmegaConf.

API#

nemo_rl.distributed.worker_group_utils.get_nsight_config_if_pattern_matches(
worker_name: str,
) → dict[str, Any][source]#

Check if worker name matches patterns in NRL_NSYS_WORKER_PATTERNS and return nsight config.

Parameters:

worker_name – Name of the worker to check against patterns

Returns:

config} if pattern matches, empty dict otherwise

Return type:

Dictionary containing {“nsight”

nemo_rl.distributed.worker_group_utils.recursive_merge_options(
default_options: dict[str, Any],
extra_options: dict[str, Any],
) → dict[str, Any][source]#

Recursively merge extra options into default options using OmegaConf.

Parameters:

  • default_options – Default options dictionary (lower precedence)

  • extra_options – Extra options provided by the caller (higher precedence)

Returns:

Merged options dictionary with extra_options taking precedence over default_options