2.11.1.dev3+gd91bf539f

NVIDIA Nsight Systems

Directory

Functions

Function Description
flyteplugins.nsight.capture_report_file() Upload the .nsys-rep and surface it as a trace output.
flyteplugins.nsight.nsys_available() True if the nsys binary is on PATH in this container.
flyteplugins.nsight.nsys_profile() Profile a Flyte task with Nsight Systems.
flyteplugins.nsight.session_name()
flyteplugins.nsight.under_nsys() True only when the runtime actually re-exec’d this process under nsys launch.
flyteplugins.nsight.nsys.profile() Profile the wrapped block as a named region.
flyteplugins.nsight.nsys.range() Profile the wrapped block as a named region.
flyteplugins.nsight.nvtx.mark() Drop a single NVTX marker at this instant.
flyteplugins.nsight.nvtx.range() Push an NVTX range on enter and pop it on exit.

Packages

Package Description
flyteplugins.nsight Flyte NVIDIA Nsight Systems plugin.
flyteplugins.nsight.nsys Region profiling: collect only part of a task.
flyteplugins.nsight.nvtx NVTX annotation helpers.