提交 9be3ec1a authored 作者: Frederic Bastien's avatar Frederic Bastien

created the fct clear_compiledir to allow python scrit to delete easily this directory.

上级 c239bfec
#!/usr/bin/env python #!/usr/bin/env python
from theano.gof import cc from theano.gof import cc
import sys, os import sys
compiledir = cc.get_compiledir() compiledir = cc.get_compiledir()
if len(sys.argv) == 1: if len(sys.argv) == 1:
print compiledir print compiledir
elif sys.argv[1] in ('clear'): elif sys.argv[1] in ('clear'):
verbose = False cc.clear_compiledir()
for l in os.listdir(compiledir):
if l.endswith('.so'):
if verbose: print 'removing', l
os.remove(os.path.join(compiledir, l))
elif l.endswith('.cpp'):
if verbose: print 'removing', l
os.remove(os.path.join(compiledir, l))
else:
if verbose: print 'skipping ', l
pass
else: else:
print 'command "%s" not recognized' % sys.argv[1] print 'command "%s" not recognized' % sys.argv[1]
sys.exit(1) sys.exit(1)
......
...@@ -35,7 +35,7 @@ from gof import \ ...@@ -35,7 +35,7 @@ from gof import \
toolbox, \ toolbox, \
Type, Generic, generic, \ Type, Generic, generic, \
object2, utils, \ object2, utils, \
set_compiledir, get_compiledir set_compiledir, get_compiledir, clear_compiledir
from compile import \ from compile import \
SymbolicInput, SymbolicInputKit, In, \ SymbolicInput, SymbolicInputKit, In, \
......
...@@ -3,7 +3,7 @@ from cc import \ ...@@ -3,7 +3,7 @@ from cc import \
CLinker, OpWiseCLinker, DualLinker CLinker, OpWiseCLinker, DualLinker
from compiledir import \ from compiledir import \
set_compiledir, get_compiledir set_compiledir, get_compiledir, clear_compiledir
from env import \ from env import \
InconsistencyError, Env InconsistencyError, Env
......
...@@ -60,3 +60,18 @@ def get_compiledir(): ...@@ -60,3 +60,18 @@ def get_compiledir():
if not hasattr(set_compiledir, 'compiledir'): if not hasattr(set_compiledir, 'compiledir'):
set_compiledir() set_compiledir()
return set_compiledir.compiledir return set_compiledir.compiledir
def clear_compiledir(verbose=False):
if not hasattr(set_compiledir, 'compiledir'):
set_compiledir()
compiledir = get_compiledir()
for l in os.listdir(compiledir):
if l.endswith('.so'):
if verbose: print 'removing', l
os.remove(os.path.join(compiledir, l))
elif l.endswith('.cpp'):
if verbose: print 'removing', l
os.remove(os.path.join(compiledir, l))
else:
if verbose: print 'skipping ', l
pass
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论