提交 bfe7bc36 authored 作者: Frederic Bastien's avatar Frederic Bastien

renamed optimizer to be more consistent with what other opt name.

上级 2a0dc002
...@@ -611,7 +611,7 @@ def gpu_print_wrapper(op, cnda): ...@@ -611,7 +611,7 @@ def gpu_print_wrapper(op, cnda):
@register_opt() @register_opt()
@local_optimizer([]) @local_optimizer([])
def local_print_op(node): def local_gpu_print_op(node):
if isinstance(node.op, tensor.printing.Print): if isinstance(node.op, tensor.printing.Print):
x, = node.inputs x, = node.inputs
if x.owner and x.owner.op == host_from_gpu: if x.owner and x.owner.op == host_from_gpu:
......
...@@ -288,7 +288,7 @@ class GpuMultinomialFromUniform(MultinomialFromUniform): ...@@ -288,7 +288,7 @@ class GpuMultinomialFromUniform(MultinomialFromUniform):
""" % locals() """ % locals()
@local_optimizer() @local_optimizer()
def use_gpu_multinomial(node): def local_gpu_multinomial(node):
if type(node.op) is MultinomialFromUniform: if type(node.op) is MultinomialFromUniform:
p, u = node.inputs p, u = node.inputs
m, = node.outputs m, = node.outputs
...@@ -309,5 +309,5 @@ def use_gpu_multinomial(node): ...@@ -309,5 +309,5 @@ def use_gpu_multinomial(node):
return [gpu_from_host(ret)] return [gpu_from_host(ret)]
if cuda_available: if cuda_available:
register_opt()(use_gpu_multinomial) register_opt()(local_gpu_multinomial)
pass pass
...@@ -12,7 +12,7 @@ def get_mode(gpu): ...@@ -12,7 +12,7 @@ def get_mode(gpu):
mode = get_default_mode() mode = get_default_mode()
mode = copy.copy(mode) mode = copy.copy(mode)
if gpu: if gpu:
mode = mode.including('gpu', 'gpu_local_optimizations', 'local_cut_gpu_host_gpu', 'use_gpu_multinomial') mode = mode.including('gpu', 'gpu_local_optimizations', 'local_cut_gpu_host_gpu', 'local_gpu_multinomial')
if isinstance(mode.linker, theano.gof.PerformLinker): if isinstance(mode.linker, theano.gof.PerformLinker):
mode.linker = predefined_linkers['c|py'] mode.linker = predefined_linkers['c|py']
return mode return mode
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论