stormlog.device_collectors

Backend-aware device memory collector abstractions.

Functions

`build_device_memory_collector`([device])	Build a backend collector for CUDA/ROCm/MPS runtime environments.
`detect_torch_runtime_backend`()	Return the active torch runtime backend in this environment.
`validate_device_memory_sample`(sample, ...[, ...])	Validate a sample against its collector's declared capabilities.

Classes

`CudaDeviceCollector`([device])	Collector for NVIDIA CUDA runtime memory counters.
`DeviceMemoryCapabilities`(backend, ...[, ...])	Immutable description of memory signals exposed by a collector.
`DeviceMemoryCollector`()	Backend-specific collector contract for device memory signals.
`DeviceMemorySample`(allocated_bytes, ...)	Normalized device-memory sample produced by a backend collector.
`DeviceMemorySampleResult`(sample[, ...])	Device-memory sample plus diagnostics about partial/core collection failures.
`MPSDeviceCollector`([device])	Collector for Apple Metal (MPS) runtime counters.
`ROCmDeviceCollector`([device])	Collector for ROCm runtimes surfaced through torch.cuda APIs.

class stormlog.device_collectors.DeviceMemoryCollector[source]

Bases: ABC

Backend-specific collector contract for device memory signals.

abstract name()[source]

Return runtime backend name (cuda, rocm, mps).

Return type:: str

abstract is_available()[source]

Return whether this collector can sample in the current runtime.

Return type:: bool

abstract sample()[source]

Collect a single normalized memory sample.

Return type:: DeviceMemorySample

sample_with_diagnostics()[source]

Collect a sample while preserving core-failure diagnostics.

Return type:: DeviceMemorySampleResult

abstract capabilities()[source]

Describe backend capability signals for telemetry metadata.

Return type:: DeviceMemoryCapabilities

class stormlog.device_collectors.DeviceMemoryCapabilities(backend, telemetry_collector, sampling_source, supports_allocator_allocated=False, supports_allocator_reserved=False, supports_allocator_active=False, supports_allocator_inactive=False, supports_device_used=False, supports_device_free=False, supports_device_total=False, supports_native_allocator_history=False, supports_fragmentation_analysis=False, supports_allocator_attribution=False, supports_bounded_profiling=False)[source]

Bases: object

Immutable description of memory signals exposed by a collector.

Parameters:

backend (str)
telemetry_collector (str)
sampling_source (str)
supports_allocator_allocated (bool)
supports_allocator_reserved (bool)
supports_allocator_active (bool)
supports_allocator_inactive (bool)
supports_device_used (bool)
supports_device_free (bool)
supports_device_total (bool)
supports_native_allocator_history (bool)
supports_fragmentation_analysis (bool)
supports_allocator_attribution (bool)
supports_bounded_profiling (bool)

backend: str

telemetry_collector: str

sampling_source: str

supports_allocator_allocated: bool = False

supports_allocator_reserved: bool = False

supports_allocator_active: bool = False

supports_allocator_inactive: bool = False

supports_device_used: bool = False

supports_device_free: bool = False

supports_device_total: bool = False

supports_native_allocator_history: bool = False

supports_fragmentation_analysis: bool = False

supports_allocator_attribution: bool = False

supports_bounded_profiling: bool = False

classmethod from_mapping(value)[source]

Convert legacy capability mappings into the typed contract.

Parameters:: value (Mapping[str, Any])
Return type:: DeviceMemoryCapabilities

to_metadata()[source]

Return a JSON-safe capability mapping for telemetry metadata.

Return type:: Dict[str, Any]

class stormlog.device_collectors.DeviceMemorySample(allocated_bytes, reserved_bytes, used_bytes, free_bytes, total_bytes, active_bytes, inactive_bytes, device_id)[source]