提交 0a7b12af authored 作者: Frederic Bastien's avatar Frederic Bastien

Small fix following code review

上级 9c89212a
...@@ -40,13 +40,12 @@ class T_bunch_of_modes(unittest.TestCase): ...@@ -40,13 +40,12 @@ class T_bunch_of_modes(unittest.TestCase):
# there should be # there should be
# - VM_Linker # - VM_Linker
# - OpWiseCLinker (FAST_RUN) # - OpWiseCLinker (FAST_RUN)
# - WrapLinker
# - PerformLinker (FAST_COMPILE) # - PerformLinker (FAST_COMPILE)
# - DebugMode's Linker (DEBUG_MODE) # - DebugMode's Linker (DEBUG_MODE)
assert 4 == len(set(linker_classes_involved)) assert 4 == len(set(linker_classes_involved))
class T_WrapLinker(unittest.TestCase): class T_old_problem(unittest.TestCase):
def test_1(self): def test_1(self):
# Then, build a mode with the same linker, and a modified optimizer # Then, build a mode with the same linker, and a modified optimizer
default_mode = theano.compile.mode.get_default_mode() default_mode = theano.compile.mode.get_default_mode()
......
...@@ -463,7 +463,7 @@ del param ...@@ -463,7 +463,7 @@ del param
if rc == 0 and config.cxx != "": if rc == 0 and config.cxx != "":
# Keep the default linker the same as the one for the mode FAST_RUN # Keep the default linker the same as the one for the mode FAST_RUN
AddConfigVar('linker', AddConfigVar('linker',
"Default linker used if the theano flags mode is Mode ", "Default linker used if the theano flags mode is Mode",
EnumStr('cvm', 'c|py', 'py', 'c', 'c|py_nogc', EnumStr('cvm', 'c|py', 'py', 'c', 'c|py_nogc',
'vm', 'vm_nogc', 'cvm_nogc'), 'vm', 'vm_nogc', 'cvm_nogc'),
in_c_key=False) in_c_key=False)
...@@ -471,7 +471,7 @@ else: ...@@ -471,7 +471,7 @@ else:
# g++ is not present or the user disabled it, # g++ is not present or the user disabled it,
# linker should default to python only. # linker should default to python only.
AddConfigVar('linker', AddConfigVar('linker',
"Default linker used if the theano flags mode is Mode ", "Default linker used if the theano flags mode is Mode",
EnumStr('vm', 'py', 'vm_nogc'), EnumStr('vm', 'py', 'vm_nogc'),
in_c_key=False) in_c_key=False)
try: try:
...@@ -499,7 +499,7 @@ AddConfigVar('allow_gc', ...@@ -499,7 +499,7 @@ AddConfigVar('allow_gc',
# Keep the default optimizer the same as the one for the mode FAST_RUN # Keep the default optimizer the same as the one for the mode FAST_RUN
AddConfigVar( AddConfigVar(
'optimizer', 'optimizer',
"Default optimizer. If not None, will use this optimizer with the Mode ", "Default optimizer. If not None, will use this optimizer with the Mode",
EnumStr('fast_run', 'merge', 'fast_compile', 'None'), EnumStr('fast_run', 'merge', 'fast_compile', 'None'),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论