def deserialize_workload_registry_entry(data): """ Deserialize a workload registry entry. This should be used along with :code:`serialize_workload_registry_entry` Parameters ---------- data: Tuple The return value of :code:`serialize_workload_registry_entry` """ global WORKLOAD_FUNC_REGISTRY name, value = data if name not in WORKLOAD_FUNC_REGISTRY: # pylint: disable=assignment-from-no-return WORKLOAD_FUNC_REGISTRY[name] = LoadJSON(value)
def __setstate__(self, state): # Since we always use tensors to recover the ComputeDAG, we do not support # (de)serialization of the ComputeDAG constructed by a schedule. self.__init_handle_by_constructor__(_ffi_api.ComputeDAG, LoadJSON(state["tensors"]), None)
def __setstate__(self, state): self.compute = LoadJSON(state["compute"]) # pylint: disable=assignment-from-no-return self.sche = LoadJSON(state["sche"]) # pylint: disable=assignment-from-no-return self.__init_handle_by_constructor__(_ffi_api.ComputeDAG, self.compute, self.sche)