setup filename for mpi

tetrarun.py

@@ -70,7 +70,7 @@ def main(argv):
       mtx_cpu = np.zeros(res.size, dtype=np.int8).reshape(res.shape)
       res.get(out=mtx_cpu)
    
-      #writetofile(outputfile+'.full', Cx_cpu, Cy_cpu, Dx_cpu, Dy_cpu, Dz_cpu, mtx_cpu, mpi)
+      writetofile(outputfile+'.full', Cx_cpu, Cy_cpu, Dx_cpu, Dy_cpu, Dz_cpu, mtx_cpu, mpi)
       #writetofile2(outputfile, Cx_cpu, Cy_cpu, Dx_cpu, Dy_cpu, Dz_cpu, mtx_cpu, mpi)
 
    if outputfile and binary:

utils.py

@@ -52,7 +52,8 @@ def writetofile(filename, Cx_cpu, Cy_cpu, Dx_cpu, Dy_cpu, Dz_cpu, mtx_cpu, mpi):
     pos = size_C * size_D
 
     if mpi:
-        filename = f'R{rank}/{size}_' + filename
+        filename = filename + f'R{rank}-{size}.out'
+    print("Filename: ", filename)
     f = open(filename, "w")
 
     for i in range(0, pos):
@@ -77,8 +78,9 @@ def writetofile2(filename, Cx_cpu, Cy_cpu, Dx_cpu, Dy_cpu, Dz_cpu, mtx_cpu, mpi)
     pos = size_C * size_D
 
     if mpi:
-        filename = f'R{rank}/{size}_' + filename
+        filename = filename + f'R{rank}-{size}.out'
 
+    print("Filename: ", filename)
     f = open(filename, "w")
     for i in range(0, pos):
         s = 0