pytorch/test/cpp_extensions
Edward Yang b4a35632f9 Add function to materialize COW storages (#117053)
Summary: From Kurt Mohler, see https://github.com/pytorch/pytorch/pull/113396 (manually imported due to ghimport problems)

Test Plan: sandcastle, OSS CI

Differential Revision: D52610522

Pull Request resolved: https://github.com/pytorch/pytorch/pull/117053
Approved by: https://github.com/malfet, https://github.com/kurtamohler
2024-01-10 15:34:16 +00:00
..
no_python_abi_suffix_test
self_compiler_include_dirs_test
torch_test_cpp_extension
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
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_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 Support large negative SymInt (#99157) 2023-04-15 22:43:51 +00:00
identity.cpp [functorch] disable C++ Function under functorch transforms (#103957) 2023-06-23 11:01:44 +00:00
jit_extension.cpp
jit_extension2.cpp
mps_extension.mm [MPS] Prerequisite for MPS C++ extension (#102483) 2023-06-07 17:28:31 +00:00
open_registration_extension.cpp Add function to materialize COW storages (#117053) 2024-01-10 15:34:16 +00:00
ort_extension.cpp Make dispatcher registrations of SymInt functions backwards compatible (#84557) 2022-09-07 16:30:21 +00:00
rng_extension.cpp Reland of "Python binding to set/get CUDA rng state offset" (#99565) 2023-04-20 15:42:25 +00:00
setup.py [MPS] Prerequisite for MPS C++ extension (#102483) 2023-06-07 17:28:31 +00:00
torch_library.cu