提交 cd0bde88 authored 作者: Frederic's avatar Frederic 提交者: AlOa

Add -N param to the second script.

上级 a06016e8
import os import os
import subprocess import subprocess
import sys import sys
from optparse import OptionParser
import theano
def runScript(): parser = OptionParser(usage='%prog <options>\n Compute time for'
' fast and slow elemwise operations')
parser.add_option('-N', '--N', action='store', dest='N',
default=theano.config.openmp_elemwise_minsize, type="int",
help="Number of vector element")
def runScript(N):
script = 'elemwise_time_test.py' script = 'elemwise_time_test.py'
dir = os.path.dirname(os.path.abspath(__file__)) dir = os.path.dirname(os.path.abspath(__file__))
proc = subprocess.Popen(['python', script, '--script'], stdout=subprocess.PIPE, stderr = subprocess.PIPE, cwd = dir) proc = subprocess.Popen(['python', script, '--script', '-N', str(N)],
stdout=subprocess.PIPE, stderr=subprocess.PIPE,
cwd=dir)
(out, err) = proc.communicate() (out, err) = proc.communicate()
if err: if err:
print err print err
...@@ -14,10 +25,14 @@ def runScript(): ...@@ -14,10 +25,14 @@ def runScript():
return map(float, out.split(" ")) return map(float, out.split(" "))
if __name__ == '__main__': if __name__ == '__main__':
options, arguments = parser.parse_args(sys.argv)
if hasattr(options, "help"):
print options.help
sys.exit(0)
os.environ['THEANO_FLAGS'] = 'openmp=false' os.environ['THEANO_FLAGS'] = 'openmp=false'
(cheapTime, costlyTime) = runScript() (cheapTime, costlyTime) = runScript(N=options.N)
os.environ['THEANO_FLAGS'] = 'openmp=true' os.environ['THEANO_FLAGS'] = 'openmp=true'
(cheapTimeOpenmp, costlyTimeOpenmp) = runScript() (cheapTimeOpenmp, costlyTimeOpenmp) = runScript(N=options.N)
if cheapTime > cheapTimeOpenmp: if cheapTime > cheapTimeOpenmp:
cheapSpeed = (cheapTime - cheapTimeOpenmp) / cheapTime cheapSpeed = (cheapTime - cheapTimeOpenmp) / cheapTime
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论