pytorch/test/cpp_extensions
Bug Hunter Yan 73dd6f04c9 extend serialization for tensor metadata (#99808)
Fixes #ISSUE_NUMBER
Add the serialization logic of backend metadata to the serialization of tensor, which is implemented through custom registration functions.

In #97429 , the structure backendMeta is provided in TensorImpl, and we think that this part of information may also need to be serialized for custom.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/99808
Approved by: https://github.com/ezyang
2023-05-04 20:32:11 +00:00
..
no_python_abi_suffix_test
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 Support large negative SymInt (#99157) 2023-04-15 22:43:51 +00:00
jit_extension.cpp
jit_extension2.cpp
open_registration_extension.cpp extend serialization for tensor metadata (#99808) 2023-05-04 20:32:11 +00:00
ort_extension.cpp
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 Fix shoud/shoudl typos (#97930) 2023-03-30 08:27:16 +00:00
torch_library.cu