loopyt 28fb291db1 style(sm90): satisfy ruff and add the missing type annotations ... No behaviour change. `CONTRIBUTING.md` requires `ruff check --no-cache` and `ruff format --check --no-cache` to pass, and the SM90 branch predates that requirement: 72 findings, all in the files added here. Most of it is mechanical (`ruff format`), plus: - annotations for the helper signatures (`torch.Tensor` / `int` / `Generator` / return types); hooks are typed as `Callable[[nn.Module, tuple, object], None]` rather than left bare; - two `if`/`else` assignments become ternaries (SIM108), and the index read in `h20_real_requant_check.py` uses a context manager (SIM115). Also corrects the `sparse_mla_fwd` docstring, which still described the old `0 <= idx <= q_position` masking contract. Passing `is_causal=True` there is what produced the all-NaN layer 0: indices address the assembled `[swa cache | current swa | compressed]` buffer, so the row number is not a valid causal bound and the engine must pass False.		2026-10-21 17:09:08 +08:00
..
__init__.py	initial commit	2026-10-20 12:09:42 +08:00
attn.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
builder.py	initial commit	2026-10-20 12:09:42 +08:00
compress.py	initial commit	2026-10-20 12:09:42 +08:00
containers.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
cross_layer.py	initial commit	2026-10-20 12:09:42 +08:00
engram_metadata.py	initial commit	2026-10-20 12:09:42 +08:00
engram_parameters.py	initial commit	2026-10-20 12:09:42 +08:00
engram.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
hc_blocks.py	initial commit	2026-10-20 12:09:42 +08:00
hc_lazy.py	initial commit	2026-10-20 12:09:42 +08:00
hc.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
heads.py	initial commit	2026-10-20 12:09:42 +08:00
index_attn.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
linear.py	initial commit	2026-10-20 12:09:42 +08:00
mhc_modeling.py	initial commit	2026-10-20 12:09:42 +08:00
mhc_ops.py	initial commit	2026-10-20 12:09:42 +08:00
mla_attn.py	initial commit	2026-10-20 12:09:42 +08:00
mlp.py	fix(sm90): expand expert scales on the contraction axis only	2026-10-21 16:50:31 +08:00
moe.py	initial commit	2026-10-20 12:09:42 +08:00
norm.py	initial commit	2026-10-20 12:09:42 +08:00
rope.py	initial commit	2026-10-20 12:09:42 +08:00
utils.py	initial commit	2026-10-20 12:09:42 +08:00
yoco.py	initial commit	2026-10-20 12:09:42 +08:00