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

set cache version in nnet ops

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