Changeset 1665 for XIOS/dev/dev_trunk_omp/src/test/test_remap_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_remap_omp.f90 (modified) (2 diffs)

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

@@ -41 +41 @@
   DOUBLE PRECISION, PARAMETER :: missing_value = 100000
   INTEGER :: 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 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
 
   CALL MPI_INIT_THREAD(3, provided, ierr)
@@ -51 +69 @@
   CALL MPI_COMM_RANK(MPI_COMM_WORLD,rank,ierr)
   CALL MPI_COMM_SIZE(MPI_COMM_WORLD,size,ierr)
-  if(rank < size-2) then
+  if(rank < size-nb_servers) then
  
   !$omp parallel default(firstprivate)