Merge pull request #19 from mathias77515/Mathias-dev

Mathias dev

src/FMM/__init__.py

@@ -1,2 +1 @@
-from .pipeline import *
-print('importing pipeline fmm')
+from .pipeline import *

src/__init__.py

@@ -1,4 +1,3 @@
 from .FMM import *
 from .CMM import *
-from .lib import *
-from .running_scripts import *
+from .lib import *

src/run.py

@@ -1,15 +1,17 @@
-#import sys
-#from pyoperators import *
-#from FMM.pipeline import PipelineEnd2End
-#from CMM.pipeline import Pipeline
-import running_scripts
+import sys
+from pyoperators import *
+from FMM.pipeline import PipelineEnd2End
+from CMM.pipeline import Pipeline
+
+### Common MPI arguments
+comm = MPI.COMM_WORLD
 stop
+
 simu = 'CMM'
 
 if __name__ == "__main__":
 
-    ### Common MPI arguments
-    comm = MPI.COMM_WORLD
+
 
 
     if simu == 'FMM':

src/run.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+
+#SBATCH --job-name=FMM
+
+# we ask for n MPI tasks with N cores each on c nodes
+
+#SBATCH --partition=htc
+#SBATCH --nodes=1                # c
+#SBATCH --ntasks-per-node=1      # n
+#SBATCH --cpus-per-task=4        # N
+#SBATCH --mem=60G
+#SBATCH --time=3-00:00:00
+#SBATCH --output=mulitple_jobs_%j.log
+#SBATCH --array=1-500
+
+export OMP_NUM_THREADS=${SLURM_CPUS_PER_TASK}
+
+module load mpich
+
+mpirun -np $SLURM_NTASKS python src/run_fmm.py