wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-10 02:41:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Re-Revert "[Core] zero-copy serializer for pytorch (#12344)" (#12478)" (#12515)

Browse Source 
This reverts commit 3f22448834.
This commit is contained in:
Amog Kamsetty
2020-11-30 19:05:55 -08:00
committed by GitHub
parent 8223a33bff
commit f9a99f20dd
5 changed files with 1 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serialization.py
-2
View File
@@ -26,7 +26,6 @@ from ray._raylet import (
MessagePackSerializedObject,
RawSerializedObject,
)
from ray import serialization_addons
logger = logging.getLogger(__name__)
@@ -156,7 +155,6 @@ class SerializationContext:
# Because objects have default __reduce__ method, we only need to
# treat ObjectRef specifically.
self._register_cloudpickle_reducer(ray.ObjectRef, object_ref_reducer)
serialization_addons.apply(self)
def _register_cloudpickle_reducer(self, cls, reducer):
pickle.CloudPickler.dispatch[cls] = reducer