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

Do not pickle functions of Blockwise

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