Skip to content

SwinUNETR: optional flash attention (scaled_dot_product_attention) in WindowAttention#8977

Open
aymuos15 wants to merge 2 commits into
Project-MONAI:devfrom
aymuos15:perf/swinunetr-flash-attention
Open

SwinUNETR: optional flash attention (scaled_dot_product_attention) in WindowAttention#8977
aymuos15 wants to merge 2 commits into
Project-MONAI:devfrom
aymuos15:perf/swinunetr-flash-attention

Add SwinUNETR flash attention equivalence test

1713e89
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 2, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO