Fix the signature of paged_attention by marking megacore_mode optional#8615

Merged

lsy323 merged 1 commit intopytorch:masterfrom fenghuizhang:master

Jan 23, 2025

+35-32

Commits on Jan 23, 2025

Fix the signature of paged_attention by marking megacore_mode optional and adds tests coverage for both megacore_mode and attn_logits_soft_cap in the dynamo code path
fenghuizhang
committed