mstar.utils#
Modules
Fused AdaRMS normalisation + scale/shift/gate Triton kernel. |
|
The user's cache root, which M*'s on-disk caches (compiled kernels, downloaded checkpoints) live under. |
|
Client-visible text for an exception raised while serving a request. |
|
Small generic containers with no engine or worker knowledge |
|
The CPUs this process may use, and the plan every process of a deployment reads. |
|
Words for a child process's exit status, for log lines and error messages. |
|
Tokenize a long prompt as parallel chunks, byte-identical to one encode. |
|
FP8 block-scale checkpoints: the |
|
Move a process's set-up heap out of the garbage collector's reach. |
|
The client-facing generation knobs, checked once where every endpoint submits. |
|
Copy-only host-to-device staging, for work that runs beside a live CUDA graph. |
|
Resolving a Hugging Face repo id to a local snapshot directory. |
|
Large host copies split over threads of their own. |
|
Engine knobs read by more than one package. |
|
Watchdog that leaves once the process that spawned this one is gone. |
|
Utilities for NVTX range annotations for profiling with nsys. |
|
Default IPC socket prefixes for the server entry points. |
|
CUDA stream state a worker has to manage. |
|
Starting threads, and moving the threads a step passes through onto their CPUs. |
|
Decode a model's text chunks without splitting characters. |