mstar.utils

mstar.utils#

Modules

adarms_norm

Fused AdaRMS normalisation + scale/shift/gate Triton kernel.

containers

Small generic containers with no engine or worker knowledge

exitcode

Words for a child process's exit status, for log lines and error messages.

h2d

Copy-only host-to-device staging, for work that runs beside a live CUDA graph.

ipc_format

logging_config

orphan

Watchdog that leaves once the process that spawned this one is gone.

profiler

Utilities for NVTX range annotations for profiling with nsys.

streams

CUDA stream state a worker has to manage.