提交 c0e864dd authored 作者: Martin Arjovsky's avatar Martin Arjovsky 提交者: AdeB

config stuff

上级 be629ae2
......@@ -6,7 +6,6 @@ import textwrap
import re
import theano
from theano import config
from theano.configparser import (AddConfigVar, BoolParam, ConfigParam, EnumStr,
FloatParam, IntParam, StrParam,
TheanoConfigParser)
......@@ -320,9 +319,9 @@ AddConfigVar('dnn.conv.precision',
def default_dnn_path(suffix):
def f(suffix=suffix):
if config.cuda.root == '':
if theano.config.cuda.root == '':
return ''
return os.path.join(config.cuda.root, suffix)
return os.path.join(theano.config.cuda.root, suffix)
return f
AddConfigVar('dnn.include_path',
......@@ -931,7 +930,8 @@ AddConfigVar('DebugMode.patience',
AddConfigVar('DebugMode.check_c',
"Run C implementations where possible",
BoolParam(bool(theano.config.cxx)),
BoolParam(
lambda: bool(theano.config.cxx)),
in_c_key=False)
AddConfigVar('DebugMode.check_py',
......@@ -1173,7 +1173,7 @@ AddConfigVar('compile.wait',
def _timeout_default():
return config.compile.wait * 24
return theano.config.compile.wait * 24
AddConfigVar('compile.timeout',
"""In seconds, time that a process will wait before deciding to
......@@ -1287,7 +1287,7 @@ AddConfigVar("compiledir_format",
in_c_key=False)
def default_compiledirname():
formatted = config.compiledir_format % compiledir_format_dict
formatted = theano.config.compiledir_format % compiledir_format_dict
safe = re.sub("[\(\)\s,]+", "_", formatted)
return safe
......@@ -1376,13 +1376,17 @@ AddConfigVar(
allow_override=False),
in_c_key=False)
def default_compiledir():
return os.path.join(
theano.config.base_compiledir,
default_compiledirname())
AddConfigVar(
'compiledir',
"platform-dependent cache directory for compiled modules",
ConfigParam(
os.path.join(
config.base_compiledir,
default_compiledirname()),
default_compiledir,
filter=filter_compiledir,
allow_override=False),
in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论