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
..
_coreml
_nnapi
cpu	Extending the Pytorch vec backend for SVE (ARM) (#119571)	2024-09-18 18:59:10 +00:00
cuda	Add option to configure reduced precision math backend for SDPA (#135964)	2024-09-24 07:11:38 +00:00
cudnn
cusparselt
mha
mkl
mkldnn
mps
nnpack
openmp
opt_einsum
quantized
xeon
xnnpack
__init__.py