pytorch/torch/csrc/jit/serialization
Zhengxu Chen 59b3f5911d [sigmoid] Support custom obj deserialization. (#133463)
Summary:
It seems we have multiple places deserializing torchbind objects. Moving the code around so that every load essentially share the same implementation.

Also added a test case "package_reader_testing" which load back the archive file in Python and eagerly validate the numerical result.

Test Plan: buck test mode/opt sigmoid/inference/test:e2e_test_cpu

Reviewed By: SherlockNoMad

Differential Revision: D61235770

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133463
Approved by: https://github.com/ydwu4
2024-08-15 17:58:44 +00:00
..
callstack_debug_info_serialization.cpp
callstack_debug_info_serialization.h
export_bytecode.cpp
export_bytecode.h
export_module.cpp
export.cpp
export.h
flatbuffer_serializer_jit.cpp
flatbuffer_serializer_jit.h
flatbuffer_serializer.cpp
flatbuffer_serializer.h
import_export_constants.h
import_export_functions.h
import_export_helpers.cpp
import_export_helpers.h
import_read.cpp
import_read.h
import_source.cpp Validate that node TK_ASSIGN have field initialized (#127878) 2024-08-14 15:27:58 +00:00
import_source.h
import.cpp
import.h
mobile_bytecode_generated.h
mobile_bytecode.fbs
onnx.cpp
onnx.h
pickle.cpp [sigmoid] Support custom obj deserialization. (#133463) 2024-08-15 17:58:44 +00:00
pickle.h [sigmoid] Support custom obj deserialization. (#133463) 2024-08-15 17:58:44 +00:00
pickler.cpp
pickler.h
python_print.cpp
python_print.h
source_range_serialization_impl.h
source_range_serialization.cpp
source_range_serialization.h
storage_context.h
type_name_uniquer.cpp
type_name_uniquer.h
unpickler.cpp
unpickler.h