提交 642ffda6 authored 作者: James Bergstra's avatar James Bergstra

set cache version in nnet ops

上级 e618b15c
...@@ -168,8 +168,8 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias (Op): ...@@ -168,8 +168,8 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias (Op):
return sio.getvalue() return sio.getvalue()
def c_code_cache_version(self): def c_code_cache_version(self):
return () #return ()
return (1,0) return (2,)
class GpuCrossentropySoftmax1HotWithBiasDx (Op): class GpuCrossentropySoftmax1HotWithBiasDx (Op):
...@@ -187,7 +187,8 @@ class GpuCrossentropySoftmax1HotWithBiasDx (Op): ...@@ -187,7 +187,8 @@ class GpuCrossentropySoftmax1HotWithBiasDx (Op):
def make_node(self, dy, sm, y_idx): def make_node(self, dy, sm, y_idx):
return Apply(self, [dy, sm, y_idx],[sm.type()]) return Apply(self, [dy, sm, y_idx],[sm.type()])
def c_code_cache_version(self): def c_code_cache_version(self):
return () return (2,)
#return ()
def c_code(self, node, nodename, (dnll, sm, y_idx), (dx,), sub): def c_code(self, node, nodename, (dnll, sm, y_idx), (dx,), sub):
fail = sub['fail'] fail = sub['fail']
return """ return """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论