Configuration#

All profiling behaviour is controlled through ProfileManager.setup(). This must be called once before any regions are created. Calls made on the ProfileManager class use its process-wide default configuration. Independent ProfileManager() instances each have their own configuration and regions; see Multiple sessions.

Multiple sessions#

Instantiate a manager when two profiling sessions need to coexist. Calls must be made through the manager that should receive the event:

from scope_profiler import ProfileManager

compute_profiler = ProfileManager()
io_profiler = ProfileManager()

with compute_profiler.session(file_path="compute.h5", verbose=False):
    with io_profiler.session(file_path="io.h5", verbose=False):
        with compute_profiler.profile_region("solve"):
            solve()
        with io_profiler.profile_region("checkpoint"):
            write_checkpoint()

The class-level API remains available as the default manager and is isolated from instantiated managers. Decorators are isolated in the same way: use @compute_profiler.profile(...) to attach a function to that session.

The sessions above coexist by nesting distinct managers in one execution thread. This feature does not make profiling thread-safe: do not enter or finalize profiling sessions concurrently from Python threads. It also does not isolate process-global instrumentation backends. In particular, LIKWID owns one marker state per process, so only one overlapping manager may use use_likwid=True. Timing-only, aggregation, line-profiling, GPU-timing, and NVTX configurations retain their existing behavior.

ProfileManager.setup() parameters#

Parameter

Type

Default

Description

deactivate_profiling

bool

False

Master switch. When True, all regions become no-ops with near-zero cost.

use_likwid

bool

False

Wrap regions with LIKWID marker API calls for hardware counter collection. Requires pylikwid.

use_line_profiler

bool

False

Enable line-by-line profiling via line_profiler. See Line-by-line profiling.

use_nvtx

bool

False

Add NVTX ranges for NVIDIA Nsight tools; requires scope-profiler[nvtx].

recursive_profile

bool

False

Enable recursive nested-call profiling for all decorated functions by default.

deactivate_file_output

bool

False

When True, write no HDF5 file at all; the run stays in memory. See below.

buffer_limit

int

1024

Initial per-region buffer capacity. Buffers grow on demand, so this is a starting size, not a cap.

file_path

str

"profiling_data.h5"

Output path for the merged HDF5 file written by finalize().

output_mode

str

"auto"

MPI writer: parallel HDF5 when available, otherwise direct token-ordered writes. Accepts auto, direct, or parallel.

hdf5_compression

str or None

None

Compress timestamp, GPU-duration, and line-profile arrays with gzip, lzf, or optional zstd.

hdf5_compression_level

int or None

None

GZIP level 0–9 or Zstandard level 1–22. LZF has no level.

hdf5_chunk_size

int or None

None

Maximum events per HDF5 chunk; enables chunked partial reads even without compression.

label

str

None

Short name for the run, used by post-processing wherever a run has to be named. See below.

capture_region_source

bool

False

Record where each region is defined (see HDF5 output & post-processing from Python). Off by default; see below for its cost and how to turn it on.

aggregation_mode

bool

False

Keep only count, total, minimum, maximum, and exclusive total per region; individual timeline events are unavailable.

HDF5 compression and chunking#

The default remains contiguous and uncompressed, which minimizes write CPU cost for small profiles. Large traces can trade some CPU time for smaller files and chunk-addressable reads:

ProfileManager.setup(
    hdf5_compression="gzip",
    hdf5_compression_level=4,
    hdf5_chunk_size=65_536,
)

gzip is portable across HDF5 installations and usually gives the smallest files of the built-in filters. lzf is faster but generally compresses less. Both use HDF5’s byte-shuffle filter, which is particularly effective for nearby int64 timestamps. Zstandard is available through the optional filter plugin:

pip install "scope-profiler[compression]"
ProfileManager.setup(
    hdf5_compression="zstd",
    hdf5_compression_level=3,
    hdf5_chunk_size=65_536,
)

Readers of a Zstandard-compressed file must also have hdf5plugin installed so HDF5 can load the filter. Compression automatically makes datasets chunked; set only hdf5_chunk_size when partial reads are desired without compression. In MPI auto mode, scope-profiler uses the direct single-file writer if the parallel HDF5 library lacks the requested filter. Explicit output_mode="parallel" instead fails early with a diagnostic, so a parallel-HDF5 test cannot silently exercise the direct backend.

Enabling capture_region_source#

Every Python region records just its defining filename and line number for decorators and direct context-manager regions, using a one-time frame/code-object lookup. It neither reads nor parses source files and adds no per-call work.

Off by default, because its cost – while cheap for a typical file – is not always. Capturing a region’s source parses its defining file’s AST and walks it once per distinct file, the first time any of its regions is created – not once per region, and not on any later call. Measured cost tracks that file’s total size, not the size or number of the regions it defines:

File

Cost, 1 rank

Cost per rank, 64 ranks (shared, oversubscribed node)

Typical, a few hundred lines

< 1 ms

a few ms

~10,000 lines, ~1,000 regions (one spanning 3,000 lines)

~0.3 s

~2.9 s

The per-region text itself is cheap to extract even when huge (~0.1 ms for a 3,000-line block) – the file-wide parse dominates. Every rank pays this independently and concurrently, so on a job with more ranks than idle cores it compounds under contention; at rank counts within the idle core count (1–8 ranks, above), it stays flat. For a typical, modestly sized codebase this is negligible either way – turn it on with:

ProfileManager.setup(capture_region_source=True)

Naming a run with label#

Post-processing names a run after its output file: run_a.h5 becomes run_a in chart legends, summary headings and the JSON statistics. label overrides that with something you choose:

ProfileManager.setup(file_path="run_a.h5", label="128 ranks")

The label is stored as metadata in the output file, so it survives into every later step — scope-profiler plot, scope-profiler inspect, the plotting functions and the exporters all pick it up with no extra flags. It is especially worth setting for scaling studies, where a legend reading 128 ranks beats one reading run_a.

Reading it back, results.label is the label or None, while results.display_label is the label or the file stem — what post-processing actually prints.

Profiling modes#

Every active region records nanosecond timestamps; the remaining flags decide what it records on top of them. This strategy dispatch picks the region class once, at setup(), so there are no runtime conditionals in the hot path:

Flags

Region class

What it records

deactivate_profiling

DisabledProfileRegion

Nothing (profiling off)

(defaults)

TimeOnlyProfileRegion

Timestamps

use_likwid

FullProfileRegion

Timestamps + LIKWID

use_line_profiler

LineProfilerRegion

Timestamps + line-by-line

use_nvtx

NVTXProfileRegion

Timestamps + NVTX ranges

use_line_profiler=True takes precedence over use_likwid=True.

use_nvtx=True adds NVTX annotations while retaining the normal CPU timing records. NVTX does not measure GPU kernel duration itself; use NVIDIA Nsight Systems/Compute or CUDA events for device-side timings. use_likwid=True and use_nvtx=True are currently separate modes, with LIKWID taking precedence.

deactivate_file_output is not part of this dispatch: recording is identical either way, and the flag only decides whether finalize() writes the buffers out. With deactivate_file_output=True, use finalize(return_results=True) to get the recorded data back — see the Python API guide.

What is no longer configurable#

Two things used to be options and are now decided for you, because there was only ever one sensible answer:

  • The run’s start time is the moment setup() is called. It is stored as the start_time_ns metadata field and is the origin of the relative timeline in post-processing.

  • MPI is used exactly when the process was started by an MPI launcher (mpirun, mpiexec, srun, …), so a plain python script.py never imports mpi4py. Set SCOPE_PROFILER_MPI=0 or =1 in the environment to overrule the detection.

Toggling profiling at runtime#

With the default class-level manager, you can leave all instrumentation in place and simply flip the master switch:

import os
from scope_profiler import ProfileManager

ProfileManager.setup(
    deactivate_profiling=os.environ.get("DISABLE_PROFILING", "0") == "1",
)

Recursive profiling of decorated entrypoints#

Set recursive_profile=True to record Python function calls made inside decorated functions:

ProfileManager.setup(recursive_profile=True)

@ProfileManager.profile("entry")
def entry():
    compute_step()

You can override this per function with @ProfileManager.profile(..., recursive=False) or @ProfileManager.profile(..., recursive=True).

When deactivate_profiling=True, every region is a DisabledProfileRegion whose __enter__ / __exit__ / wrap are trivial no-ops, adding only the cost of a Python function call (~0.1 µs).

Re-configuring#

Calling setup() again resets all existing regions and applies the new configuration:

ProfileManager.setup(file_path="run_a.h5")
# ... profile some code ...
ProfileManager.finalize()

# Start a fresh session with different settings
ProfileManager.setup(file_path="run_b.h5", use_line_profiler=True)
# ...
ProfileManager.finalize()