提交 7c8d604f authored 作者: James Bergstra's avatar James Bergstra

tweaked NaiveAlgo to be picklable

上级 38c79505
......@@ -79,7 +79,7 @@ class GpuElemwise(Op):
self.ufunc = None
self._rehash()
self.src_generator = NaiveAlgo(**self.__dict__)
self.src_generator = NaiveAlgo(self.scalar_op)
def __getstate__(self):
d = copy.copy(self.__dict__)
......
......@@ -202,10 +202,11 @@ class RecAlgo(object):
class NaiveAlgo(object):
verbose = False
def __init__(self, **kwargs):
self.__dict__.update(kwargs)
#cache_version = ()
cache_version = ('debug', 2)
self.cache_version = ()
def __init__(self, scalar_op):
self.scalar_op = scalar_op
def c_src_kernel(self, node, nodename):
nd = node.outputs[0].type.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论