Skip to content

Phase 3 — Python coverage for atomr-accel-telemetry #3

Description

@cognect

Wrap atomr-accel-telemetry for Python so users can drive observability primitives from notebooks / scripts.

Scope

atomr-accel-telemetry

Wrap as python/atomr_accel/telemetry.py + src/telemetry.rs (gated on
the upstream telemetry cargo feature):

  • NvtxKernelTrace — NVTX kernel-range markers.
  • NvmlActor — power/temp/ECC/clock probes.
  • CuptiSession — CUPTI activity tracing.

Hooks

The CUDA crate's observability module installs an atomr-telemetry
extension on the actor system. Phase 3 should:

  1. Surface TelemetryBus.subscribe_topic(...) from upstream's
    atomr-pycore (already wrapped there as atomr.telemetry.TelemetryBus).
  2. Add Device.enable_telemetry() to install the extension at runtime.
  3. Wrap NvtxKernelTrace::span(name) as a Python context manager
    that emits a CUDA NVTX range around its body.

Non-goals

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions