rohan-tan-bhowmik e86f56bc76 [Torch] [TMTensor] Added mask and is_causal support for torch.aten.scaled_dot_product_attention (#3690) ... Enabled mask and is_causal parameters for torch.aten.scaled_dot_product attention + relevant comments + tests. The tests added highlight the new capabilities introduced in this PR, including: Attention with F16 mask Attention with Boolean mask Causal attention with same Q K V shapes Causal attention without Q K V shapes Made sure that one cannot input both mask and is_causal.		2024-09-09 15:51:41 -07:00
..
jit_ir_common	Additional information in error message (#2783)	2024-01-30 08:28:08 -08:00
ltc	[NFC reformat] Run pre-commit on all files and format misc.	2024-04-27 14:08:09 -07:00
onnx_c_importer	build: find Protobuf using config mode search (#2900)	2024-02-12 17:31:41 -06:00
pt1	[Torch] [TMTensor] Added mask and is_causal support for torch.aten.scaled_dot_product_attention (#3690)	2024-09-09 15:51:41 -07:00
CMakeLists.txt	[NFC reformat] Run pre-commit on all files and format misc.	2024-04-27 14:08:09 -07:00