pytorch/torch/csrc/jit/serialization
Sergey Zimin b99fc9d29f [MTIA] Support loading Tensors on mtia:0 for pytorch code (#149327)
Summary: The diff includes updates to the PyTorch code to enable loading tensors to MTIA.

Reviewed By: PatriceVignola

Differential Revision: D71176848

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149327
Approved by: https://github.com/ezyang
2025-03-20 18:05:15 +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
import_source.h
import.cpp
import.h
mobile_bytecode_generated.h
mobile_bytecode.fbs
onnx.cpp
onnx.h
pickle.cpp
pickle.h
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 [MTIA] Support loading Tensors on mtia:0 for pytorch code (#149327) 2025-03-20 18:05:15 +00:00
unpickler.h