Zhenbin Lin 217d328764 OpenReg: Support autograd (#140662) ...

Add some unfinished implements to support autograd.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/140662
Approved by: https://github.com/ezyang

2024-11-14 23:47:56 +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

OpenReg: Support autograd (#140662)

2024-11-14 23:47:56 +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

[2/N] [Dispatchable Collectives] Extract ProcessGroup::Work into a separate class and update references (#83680)

2022-09-14 13:05:58 +00:00

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

Resubmit #67161 (#67735)

2021-11-04 09:59:30 -07:00

cuda_dlink_extension_add.cu

Allow relocatable device code linking in pytorch CUDA extensions (#78225)

2022-06-02 21:35:56 +00:00

cuda_dlink_extension_add.cuh

Allow relocatable device code linking in pytorch CUDA extensions (#78225)

2022-06-02 21:35:56 +00:00

cuda_dlink_extension_kernel.cu

Allow relocatable device code linking in pytorch CUDA extensions (#78225)

2022-06-02 21:35:56 +00:00

cuda_dlink_extension.cpp

Allow relocatable device code linking in pytorch CUDA extensions (#78225)

2022-06-02 21:35:56 +00:00

cuda_extension_kernel2.cu

Check CUDA kernel launches (/fbcode/caffe2/) (#49145)

2020-12-10 10:43:03 -08:00

cuda_extension_kernel.cu

Check CUDA kernel launches (/fbcode/caffe2/) (#49145)

2020-12-10 10:43:03 -08:00

cuda_extension.cpp

…

cuda_extension.cu

Check CUDA kernel launches (/fbcode/caffe2/) (#49145)

2020-12-10 10:43:03 -08:00

cudnn_extension.cpp

…

cusolver_extension.cpp

Resubmit #67161 (#67735)

2021-11-04 09:59:30 -07:00

dangling_impl_extension.cpp

Add a test case for findDanglingImpls (#61104)

2021-07-07 13:34:16 -07:00

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

[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

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

2024-10-04 00:26:38 +00:00

mtia_extension.cpp

Make Context to be Device-agnostic Step by Step (1/N) (#136519) (#138155)

2024-10-17 20:58:56 +00:00

open_registration_extension.cpp

Remove redundant code (#134955)

2024-09-05 01:11:32 +00:00

rng_extension.cpp

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

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

setup.py

[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

torch_library.cu

Fix UB in library.h (#57962)

2021-05-11 16:04:02 -07:00