Newer
Older

Adrian Pope
committed
import sys
import os
import numpy as N
import decomp as D
if len(sys.argv) < 6:
print('USAGE: %s <nranks> <ngName> <exeName> <maxAspectRatio> <outBase>' % sys.argv[0])
sys.exit(-1)
nranks = int(sys.argv[1])
ngName = sys.argv[2]
exeName = sys.argv[3]
maxAspectRatio = float(sys.argv[4])
outBase = sys.argv[5]
ngArr = N.loadtxt(ngName)
for ng in ngArr:
d3ds = D.filterNgNranks(ng, nranks, maxAspectRatio)
for d3d in d3ds:
t3d = d3d[1]
tmpBase = '%s_%d_%d_%d_%d_%d' % (outBase,nranks,ng,t3d[0],t3d[1],t3d[2])
outName = tmpBase + '.out.txt'
errName = tmpBase + '.err.txt'
datName = tmpBase + '.data.txt'
cmd = '%s %d %d %d %d %d %d %d 1> %s 2> %s' % (exeName,ng,ng,ng,nranks,t3d[0],t3d[1],t3d[2],outName,errName)
os.system(cmd)
os.system('touch error.data')
os.system('mv error.data %s' % datName )