FEI 59e4e92556 sdp::SDPBackend::flash_attention support PrivateUse1 (#126392) ...

Fixes https://github.com/pytorch/pytorch/issues/124271

cc  @cpuhrsch @drisspg @albanD @soulitzer

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126392
Approved by: https://github.com/drisspg

2024-06-28 17:48:40 +00:00

..

no_python_abi_suffix_test

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

self_compiler_include_dirs_test

…

torch_test_cpp_extension

…

cpp_c10d_extension.cpp

…

cpp_c10d_extension.hpp

…

cpp_frontend_extension.cpp

…

cublas_extension.cpp

…

cuda_dlink_extension_add.cu

…

cuda_dlink_extension_add.cuh

…

cuda_dlink_extension_kernel.cu

…

cuda_dlink_extension.cpp

…

cuda_extension_kernel2.cu

…

cuda_extension_kernel.cu

…

cuda_extension.cpp

…

cuda_extension.cu

…

cudnn_extension.cpp

…

cusolver_extension.cpp

…

dangling_impl_extension.cpp

…

doubler.h

…

extension.cpp

Support torch.dtype as parameter in pybind11 cpp extension. (#126865)

2024-05-29 23:19:32 +00:00

identity.cpp

…

jit_extension2.cpp

…

jit_extension.cpp

…

maia_extension.cpp

[codemod] c10::optional -> std::optional in caffe2/aten/src/ATen/DeviceGuard.h +117 (#126901)

2024-05-24 00:26:15 +00:00

mps_extension.mm

…

mtia_extension.cpp

…

open_registration_extension.cpp

sdp::SDPBackend::flash_attention support PrivateUse1 (#126392)

2024-06-28 17:48:40 +00:00

rng_extension.cpp

…

setup.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

torch_library.cu

…