Jianyu Huang 0a35986cdb Add option to configure reduced precision math backend for SDPA (#135964) ... Summary: Address https://github.com/pytorch/pytorch/issues/135778 by adding a global flag to configure whether using high precision or low precision for math backend of SDPA. Test Plan: buck2 run mode/opt //scripts/feikou/llm:run_attn_kernels Differential Revision: D62625515 Pull Request resolved: https://github.com/pytorch/pytorch/pull/135964 Approved by: https://github.com/jbschlosser		2024-09-24 07:11:38 +00:00
..
__init__.py	Add option to configure reduced precision math backend for SDPA (#135964)	2024-09-24 07:11:38 +00:00