提交 f8583b40 authored 作者: James Bergstra's avatar James Bergstra

Stack VM - warning re: python impl when callback is passed

上级 6c469782
""" """
VMs that run Theano graph computations. VMs that run Theano graph computations.
""" """
import logging
import sys import sys
import time import time
import link import link
...@@ -13,6 +14,8 @@ config = theano.config ...@@ -13,6 +14,8 @@ config = theano.config
from theano.configparser import config, AddConfigVar, BoolParam from theano.configparser import config, AddConfigVar, BoolParam
from theano import config from theano import config
logger = logging.getLogger(__file__)
AddConfigVar('profile', AddConfigVar('profile',
"If VM should collect profile information", "If VM should collect profile information",
BoolParam(False)) BoolParam(False))
...@@ -435,6 +438,8 @@ class VM_Linker(link.LocalLinker): ...@@ -435,6 +438,8 @@ class VM_Linker(link.LocalLinker):
pre_call_clear = [storage_map[v] for v in self.no_recycling] pre_call_clear = [storage_map[v] for v in self.no_recycling]
if self.callback is not None: if self.callback is not None:
if use_cloop:
logger.warn('CLoop does not support callback, using Stack VM.')
vm = Stack( vm = Stack(
nodes, thunks, pre_call_clear, nodes, thunks, pre_call_clear,
storage_map, compute_map, storage_map, compute_map,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论