Skip to content

[Torch] [TMTensor] Added mask and is_causal support for torch.aten.scaled_dot_product_attention#3690

Merged
rsuderman merged 31 commits intollvm:mainfrom
rohan-tan-bhowmik:sdpa_mask
Sep 9, 2024
Merged

[Torch] [TMTensor] Added mask and is_causal support for torch.aten.scaled_dot_product_attention#3690
rsuderman merged 31 commits intollvm:mainfrom
rohan-tan-bhowmik:sdpa_mask

Commits

Commits on Sep 7, 2024