Skip to content

[PyTorch] Adjust checkpointing of FP8 metadata for attention - #917

Merged
cyanguwa merged 15 commits into
NVIDIA:mainfrom
cyanguwa:basemodule_dpa
Jun 15, 2024
Merged

[PyTorch] Adjust checkpointing of FP8 metadata for attention#917
cyanguwa merged 15 commits into
NVIDIA:mainfrom
cyanguwa:basemodule_dpa

[pre-commit.ci] auto fixes from pre-commit.com hooks

2fa3f36
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs