mstar.utils#
Modules
Fused AdaRMS normalisation + scale/shift/gate Triton kernel. |
|
FlashInfer utility wrappers for batched paged attention. |
|
Utilities for NVTX range annotations for profiling with nsys. |
Modules
Fused AdaRMS normalisation + scale/shift/gate Triton kernel. |
|
FlashInfer utility wrappers for batched paged attention. |
|
Utilities for NVTX range annotations for profiling with nsys. |