mstar.engine.resources.attn

mstar.engine.resources.attn#

Modules

base

The attention resource's shared machinery: the spec-time factory, the custom op every backend attends through, and the workspace pool.

config

What a model declares about attention: its backend, its spec, its step.

cross

Cross-attention over a context written once and never extended.

dense

Attention as one dense FlashAttention-3 varlen pass.

flashinfer

Paged attention through FlashInfer's prefill/decode wrappers.

ragged

wrappers

FlashInfer utility wrappers for batched paged attention.