提交 946a8fa5 authored 作者: Frederic Bastien's avatar Frederic Bastien

make code versionable.

上级 790462b6
...@@ -25,8 +25,7 @@ class Multinomial(Op): ...@@ -25,8 +25,7 @@ class Multinomial(Op):
return [None, None] return [None, None]
def c_code_cache_version(self): def c_code_cache_version(self):
return () return (3,)
#return (2,)
def c_code(self, node, name, (pvals, unis), (z,), sub): def c_code(self, node, name, (pvals, unis), (z,), sub):
...@@ -115,7 +114,7 @@ class GpuMultinomial(Multinomial): ...@@ -115,7 +114,7 @@ class GpuMultinomial(Multinomial):
def c_code_cache_version(self): def c_code_cache_version(self):
#return () #return ()
return (1,) return (super(GpuMultinomial,self).c_code_cache_version(),1)
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
return """ return """
......
...@@ -195,9 +195,6 @@ class mrg_uniform(mrg_uniform_base): ...@@ -195,9 +195,6 @@ class mrg_uniform(mrg_uniform_base):
o_rstate[0] = node.outputs[0].type.filter(rstate) # send to GPU if necessary o_rstate[0] = node.outputs[0].type.filter(rstate) # send to GPU if necessary
o_sample[0] = node.outputs[1].type.filter(rval.reshape(size))# send to GPU if necessary o_sample[0] = node.outputs[1].type.filter(rval.reshape(size))# send to GPU if necessary
def c_code_cache_version(self):
return ()
def c_code(self, node, name, (rstate, size), (o_rstate, o_sample), sub): def c_code(self, node, name, (rstate, size), (o_rstate, o_sample), sub):
if self.inplace: if self.inplace:
o_rstate_requirement = 'NPY_C_CONTIGUOUS|NPY_ALIGNED' o_rstate_requirement = 'NPY_C_CONTIGUOUS|NPY_ALIGNED'
...@@ -447,9 +444,6 @@ class GPU_mrg_uniform(mrg_uniform_base): ...@@ -447,9 +444,6 @@ class GPU_mrg_uniform(mrg_uniform_base):
""" %locals() """ %locals()
def c_code_cache_version(self):
return ()
def c_code(self, node, nodename, (rstate, size), (o_rstate, o_sample), sub): def c_code(self, node, nodename, (rstate, size), (o_rstate, o_sample), sub):
inplace = int(self.inplace) inplace = int(self.inplace)
ndim = self.output_type.ndim ndim = self.output_type.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论