Skip to content

CSA: add fused Compressor forward+backward CuTe-DSL kernels (ported from Megatron-LM)#427

Open
zkyue wants to merge 4 commits into
NVIDIA:developfrom
zkyue:feat/csa-fused-compressor
Open

CSA: add fused Compressor forward+backward CuTe-DSL kernels (ported from Megatron-LM)#427
zkyue wants to merge 4 commits into
NVIDIA:developfrom
zkyue:feat/csa-fused-compressor

Address review feedback: docs bounds, lint, warning stacklevel

148fd06
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs