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