提交 aff313d8 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Modifications following code review of 4ba2eea78eaa.

上级 831c5b72
...@@ -12,11 +12,22 @@ import gof ...@@ -12,11 +12,22 @@ import gof
from compile import optdb from compile import optdb
from tensor import opt from tensor import opt
_logger = logging.getLogger('theano.lazycond')
def warning(*msg):
_logger.warning(_logger_name+'WARNING: '+' '.join(str(m) for m in msg))
def info(*msg):
_logger.info(_logger_name+'INFO: '+' '.join(str(m) for m in msg))
def debug(*msg):
_logger.debug(_logger_name+'DEBUG: '+' '.join(str(m) for m in msg))
@gof.local_optimizer([None]) @gof.local_optimizer([None])
def ifelse_make_inplace(node): def ifelse_make_inplace(node):
op = node.op op = node.op
if isinstance(op, IfElse) and not op.as_view : if isinstance(op, IfElse) and not op.as_view :
print 'ifelse_make_inplace applied' logger.debug('ifelse_make_inplace applied')
return IfElse(as_view = True, return IfElse(as_view = True,
gpu = op.gpu, name=op.name).make_node(*node.inputs).outputs gpu = op.gpu, name=op.name).make_node(*node.inputs).outputs
return False return False
......
...@@ -275,13 +275,11 @@ def local_gpu_dot_to_dot22(node): ...@@ -275,13 +275,11 @@ def local_gpu_dot_to_dot22(node):
@local_optimizer([]) @local_optimizer([])
def local_gpu_lazy_ifelse(node): def local_gpu_lazy_ifelse(node):
""" """
gpu_from_host(dot22) -> gpudot(gpu_from_host) gpu_from_host(ifelse) -> gpu_ifelse(gpu_from_host)
dot(host_from_gpu) -> host_from_gpu(gpudot22) ifelse(host_from_gpu) -> host_from_gpu(ifelse)
""" """
import theano if hasattr(theano, "lazycond"):
if hasattr(theano,"lazycond"):
gpu_ifelse = theano.lazycond.IfElse(gpu = True) gpu_ifelse = theano.lazycond.IfElse(gpu = True)
if node.op == gpu_from_host: if node.op == gpu_from_host:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论