Skip to content

Cache ctx.saved_tensors in Mamba3 backward to fix FSDP activation checkpointing#909

Open
anasiri wants to merge 1 commit into
state-spaces:mainfrom
anasiri:fix/checkpoint-saved-tensors-double-access
Open

Cache ctx.saved_tensors in Mamba3 backward to fix FSDP activation checkpointing#909
anasiri wants to merge 1 commit into
state-spaces:mainfrom
anasiri:fix/checkpoint-saved-tensors-double-access

Cache ctx.saved_tensors to avoid double-access under activation check…

5de28fe
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs