zeshengzong 77354e22e1 [OpenReg] Add AMP Integration guide for accelerators (#162050) ...

Fix part of #158917

Add AMP integration document and OpenReg code as example to explain steps of integration.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162050
Approved by: https://github.com/albanD

Co-authored-by: FFFrog <ljw1101.vip@gmail.com>

2025-09-30 12:27:11 +00:00

..

libtorch_agnostic_extension

[STABLE ABI] Add clone method to torch::stable::Tensor (#161896)

2025-09-22 20:39:24 +00:00

no_python_abi_suffix_test

[BE][Easy][7/19] enforce style for empty lines in import segments in test/[a-c]*/ and test/[q-z]*/ (#129758)

2024-07-31 10:54:03 +00:00

open_registration_extension/torch_openreg

[OpenReg] Add AMP Integration guide for accelerators (#162050)

2025-09-30 12:27:11 +00:00

python_agnostic_extension

Replace setup.py bdist_wheel with python -m build --wheel (#156712)

2025-09-29 21:51:32 +00:00

self_compiler_include_dirs_test

…

torch_stable_test_extension

Error when TORCH_STABLE_ONLY is defined in TensorBase.h (#161658)

2025-08-28 04:36:31 +00:00

torch_test_cpp_extension

[RFC] Add support for device extension autoloading (#127074)

2024-07-09 06:14:13 +00:00

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

[functorch] disable C++ Function under functorch transforms (#103957)

2023-06-23 11:01:44 +00:00

jit_extension2.cpp

…

jit_extension.cpp

…

maia_extension.cpp

[MAIA] [Autocast] Enable autocast on MAIA device (#148511)

2025-03-18 03:46:22 +00:00

mps_extension.mm

[MPSHooks] Release pending command encoder (#164093)

2025-09-29 17:50:12 +00:00

mtia_extension.cpp

Fix mtia_extension.cpp setDevice() to correctly set current_device (#149398)

2025-03-31 06:07:22 +00:00

open_registration_extension.cpp

[OpenReg][1/N] Migrate cpp_extensions_open_device_registration to OpenReg (#156588)

2025-06-26 03:59:50 +00:00

rng_extension.cpp

[codemod] c10:optional -> std::optional (#126135)

2024-05-14 19:35:51 +00:00

setup.py

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00

torch_library.cu

…

xpu_extension.sycl

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00