Mikayla Gawarecki e4e4dbd2f8 Add beginnings of torch::stable::accelerator (#159679) ...

Adds
- `torch::stable::accelerator::DeviceGuard`: `std::unique_ptr` to `DeviceGuardOpauqe` mostly copied from the below (but made generic)

   50eac811a6/torch/csrc/inductor/aoti_runtime/utils_cuda.h (L30-L46)
    - constructor `DeviceGuard(DeviceIndex)` (**this matches aoti but defers from the actual c10 DeviceGuard constructor that takes in device**)
    - `set_index(DeviceIndex)`
- `torch::stable::accelerator::Stream`: `std::shared_ptr` to `StreamOpaque`
     - constructor `Stream(StreamHandle stream)` (similar to torch::stable::Tensor)
     - `id() -> StreamId`

- `getCurrentStream(DeviceIndex device_index) -> stable::accelerator::Stream`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159679
Approved by: https://github.com/guangyey, https://github.com/janeyx99

2025-08-13 23:42:24 +00:00

..

libtorch_agnostic_extension

Add beginnings of torch::stable::accelerator (#159679)

2025-08-13 23:42: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] Disable automatic inclusion of data files (#159845)

2025-08-06 10:35:13 +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

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

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

cpp_c10d_extension.cpp

…

cpp_c10d_extension.hpp

Refactor distribuetd to use absolute header path (#85780)

2022-09-30 05:13:50 +00:00

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

[BE] Use helper functions in mps_extension (#137313)

2024-10-04 00:26:38 +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