Changeset 1665 for XIOS/dev/dev_trunk_omp/src/test/test_omp.f90

Timestamp:

05/20/19 13:14:51 (5 years ago)

Author:

yushan

Message:

MARK: branch merged with trunk @1660. Add option --omp to enable multithreading.

File:

• XIOS/dev/dev_trunk_omp/src/test/test_omp.f90 (modified) (2 diffs)

XIOS/dev/dev_trunk_omp/src/test/test_omp.f90

@@ -33 +33 @@
   INTEGER :: i,j,l,ts,n, provided
 
+  integer :: num_args, ix, nb_servers
+  character(len=12), dimension(:), allocatable :: args
+
+  num_args = command_argument_count()
+  if(num_args<1) then
+    print*, "please give the number of servers as input argument."
+    call abort
+  endif
+
+  allocate(args(num_args))  ! I've omitted checking the return status of the allocation 
+
+  do ix = 1, num_args
+    call get_command_argument(ix,args(ix))
+    ! now parse the argument as you wish
+  end do
+
+  READ(args(1),*) nb_servers
+
 !!! MPI Initialization   
 
@@ -45 +63 @@
     CALL MPI_COMM_RANK(MPI_COMM_WORLD,rank,ierr)
     CALL MPI_COMM_SIZE(MPI_COMM_WORLD,size,ierr)
-    if(rank < size-4) then
+    if(rank < size-nb_servers) then
 
     !$omp parallel default(firstprivate)