Skip to content

[PyTorch] Fix FP8 logic related to FA2/FA3 - #1141

Merged
cyanguwa merged 5 commits into
NVIDIA:mainfrom
cyanguwa:fix_fa2_logic_after_fa3
Aug 30, 2024
Merged

[PyTorch] Fix FP8 logic related to FA2/FA3#1141
cyanguwa merged 5 commits into
NVIDIA:mainfrom
cyanguwa:fix_fa2_logic_after_fa3

prefer fused attn for FP8

941e8f9
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs