FFFrog 1b389025ba Refactor and Improve the OpenReg Module (#158090) ... ---- # Refactor and Improve the OpenReg Module ## Background Since PrivateUse1 has become the main path for integrating new devices with PyTorch, there have been some feature requests related to PrivateUse1 regarding interfaces, documentation, reference examples, etc., such as the following: - https://github.com/pytorch/pytorch/issues/155864 - https://github.com/pytorch/pytorch/issues/144955 - https://github.com/pytorch/pytorch/issues/144845 Taking these requests into consideration and combining them with the position of OpenReg, which is currently used as the test backend for PrivateUse1, I'm planning to make the following optimizations: - Optimize the implementation of OpenReg to make it align with the standard specifications for real backend (C++) access, serving as a reference for new device integration code. - Add comprehensive documentation to the [developer notes](https://docs.pytorch.org/docs/main/notes.html) to guide new accelerator integration, functioning as a reference manual. ## Design Principles: - Minimization Principle: Keep the code small and clear; only implement the minimum set of code required for verification and as an integration reference. - Authenticity Principle: Integrate OpenReg in the same way that real accelerators access PyTorch. ## More Infos: Pleaes refer to [this](6b8020f1ab/test/cpp_extensions/open_registration_extension/torch_openreg/README.md) for more information about `OpenReg`. ## Current Progress: - Refer to the implementation of [torch_xla](https://github.com/pytorch/xla) to refactor all of OpenReg's code, making it easier to understand. - Ensure all tests in [test/test_openreg.py](https://github.com/FFFrog/pytorch/blob/openreg/test/test_openreg.py) pass after refactoring. ## Next Steps: - Add more features to cover all integration points. - Gradually add user guides and documentation to the [developer notes](https://docs.pytorch.org/docs/main/notes.html). Pull Request resolved: https://github.com/pytorch/pytorch/pull/158090 Approved by: https://github.com/seemethere, https://github.com/albanD		2025-07-15 08:10:05 +00:00
..
libtorch_agnostic_extension	Add a basic shim and stable::Tensor is_contiguous API (#156228)	2025-06-20 17:59:52 +00:00
no_python_abi_suffix_test
open_registration_extension/torch_openreg	Refactor and Improve the OpenReg Module (#158090)	2025-07-15 08:10:05 +00:00
python_agnostic_extension	Make python_agnostic cpp extension tests standalone (#153274)	2025-05-20 19:18:09 +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_kernel.cu
cuda_extension_kernel2.cu
cuda_extension.cpp
cuda_extension.cu
cudnn_extension.cpp
cusolver_extension.cpp
dangling_impl_extension.cpp
doubler.h
extension.cpp
identity.cpp
jit_extension.cpp
jit_extension2.cpp
maia_extension.cpp	[MAIA] [Autocast] Enable autocast on MAIA device (#148511)	2025-03-18 03:46:22 +00:00
mps_extension.mm
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
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