U ʹj @sUddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddZ dd Zd d ZiZeed <iZeed <iZeed<iZeed<dS)N)ImporterOrderedImporterPackageImporter sys_importer)create_pickler)PackageUnpickler)_maybe_decode_asciicsggt|tjjr|nd}|dk r2t|tntfdd}t}t|}||_ | || }||r|j ndfS)Ncst|st|tjjrbt|tjjr4|j}|j}n |}tj}||dt dfSt |drt t |dkrdt |f|t t |<t t |SdS)Nstorage__reduce_deploy__ reduce_deploy)torchZ is_storage isinstancer TypedStorageZ_untyped_storagedtypeZuint8appendlenhasattr_serialized_reducesgetidr )objr r importersserialized_dtypesserialized_storagesL/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_deploy.py persistent_ids$     z%_save_storages..persistent_id) rr packagerrrioBytesIOrrdumpgetvalue zip_reader)importerrrZdata_bufZpicklerZ data_valuerrr_save_storages s"    r&c sVfdd}dk r(ttt}nt}t|t|}||_|}t|<|S)Ncst|tstt|d}|dd}|dkrZ|d}|d}tjj||dS|dkr|\}}}|tkr|t f|t|<t|SdS)Nrr r )Z wrap_storagerr ) rtupleAssertionErrorrr r rZuntyped_loaded_reduces _raw_packages)Zsaved_idtypenamedatar rZ reduce_idfuncargsrrr$rrpersistent_load>s      z'_load_storages..persistent_load) r _get_packagerrr r!r0load_deploy_objects) rr$Z obj_bytesrrr0r%Z unpicklerresultrr/r_load_storages=sr5cCs|tkrt|t|<t|S)N)r*r)r$rrrr1`s r1r*r3rr))r r Z torch.packagerrrrZtorch.package._package_picklerrZ torch.package._package_unpicklerrZtorch.serializationrr&r5r1r*dict__annotations__r3rr)rrrrs   3#