提交 f8c0c4df authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Do not pickle functions of Blockwise

上级 7b7ab9e7
from collections.abc import Sequence
from copy import copy
from typing import Any, Optional, cast
import numpy as np
......@@ -87,6 +88,11 @@ class Blockwise(Op):
self._gufunc = None
super().__init__(**kwargs)
def __getstate__(self):
d = copy(self.__dict__)
d["_gufunc"] = None
return d
def _create_dummy_core_node(self, inputs: Sequence[TensorVariable]) -> Apply:
core_input_types = []
for i, (inp, sig) in enumerate(zip(inputs, self.inputs_sig)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论