from rsf.proj import *

if not WhereIs('mpirun'):
    sys.stderr.write("\nNo MPI.\n")
    sys.exit(1)

NP = int(ARGUMENTS.get('NP','3'))

Flow('a',None,'math n1=512 n2=256 output="x1+x2"')
Flow('b',None,'math n1=512 n2=256 output="-x1-x2"')

Flow('c','a b',
     '''
     mpihello input=${SOURCES[1]} b=${SOURCES[2]}
     ''',stdin=0,np=NP)

End()