nemo_rl.utils.nsys

Contenu du Module

Nom	Description
`ProfilablePolicy`	Classe de base pour les classes de protocole.

Nom	Description
`maybe_gpu_profile_step`	Néant
`wrap_with_nvtx_name`	Un décorateur pour envelopper une fonction avec une plage NVTX avec le nom donné.

NRL_NSYS_WORKER_PATTERNS NRL_NSYS_PROFILE_STEP_RANGE

nemo_rl.utils.nsys.NRL_NSYS_WORKER_PATTERNS

Valeur: get(...)

nemo_rl.utils.nsys.NRL_NSYS_PROFILE_STEP_RANGE

Valeur: get(...)

class nemo_rl.utils.nsys.ProfilablePolicy

Bases: typing.Protocol

start_gpu_profiling() -> None

stop_gpu_profiling() -> None

nemo_rl.utils.nsys.maybe_gpu_profile_step(
    policy: nemo_rl.utils.nsys.ProfilablePolicy, step: int
)

nemo_rl.utils.nsys.wrap_with_nvtx_name(name: str)

Un décorateur pour envelopper une fonction avec une plage NVTX avec le nom donné.