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

make code versionable.

上级 790462b6
......@@ -25,8 +25,7 @@ class Multinomial(Op):
return [None, None]
def c_code_cache_version(self):
return ()
#return (2,)
return (3,)
def c_code(self, node, name, (pvals, unis), (z,), sub):
......@@ -115,7 +114,7 @@ class GpuMultinomial(Multinomial):
def c_code_cache_version(self):
#return ()
return (1,)
return (super(GpuMultinomial,self).c_code_cache_version(),1)
def c_support_code_apply(self, node, nodename):
return """
......@@ -247,4 +246,4 @@ def use_gpu_multinomial(node):
return [host_from_gpu(gpu_multinomial(*[gpu_from_host(i) for i in node.inputs]))]
if theano.config.device.startswith('gpu'):
register_specialize(use_gpu_multinomial)
\ No newline at end of file
......@@ -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_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):
if self.inplace:
o_rstate_requirement = 'NPY_C_CONTIGUOUS|NPY_ALIGNED'
......@@ -447,9 +444,6 @@ class GPU_mrg_uniform(mrg_uniform_base):
""" %locals()
def c_code_cache_version(self):
return ()
def c_code(self, node, nodename, (rstate, size), (o_rstate, o_sample), sub):
inplace = int(self.inplace)
ndim = self.output_type.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论