提交 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
from theano.gof import cc
import sys, os
import sys
compiledir = cc.get_compiledir()
if len(sys.argv) == 1:
print compiledir
elif sys.argv[1] in ('clear'):
verbose = False
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
cc.clear_compiledir()
else:
print 'command "%s" not recognized' % sys.argv[1]
sys.exit(1)
......
......@@ -35,7 +35,7 @@ from gof import \
toolbox, \
Type, Generic, generic, \
object2, utils, \
set_compiledir, get_compiledir
set_compiledir, get_compiledir, clear_compiledir
from compile import \
SymbolicInput, SymbolicInputKit, In, \
......
......@@ -3,7 +3,7 @@ from cc import \
CLinker, OpWiseCLinker, DualLinker
from compiledir import \
set_compiledir, get_compiledir
set_compiledir, get_compiledir, clear_compiledir
from env import \
InconsistencyError, Env
......
......@@ -60,3 +60,18 @@ def get_compiledir():
if not hasattr(set_compiledir, 'compiledir'):
set_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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论