/[MITgcm]/MITgcm/model/src/main_do_loop.F
ViewVC logotype

Diff of /MITgcm/model/src/main_do_loop.F

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph | View Patch Patch

revision 1.10 by heimbach, Mon Apr 29 18:16:05 2013 UTC revision 1.11 by heimbach, Thu May 30 21:07:09 2013 UTC
# Line 31  C $Name$ Line 31  C $Name$
31  CBOP  CBOP
32  C     !ROUTINE: MAIN_DO_LOOP  C     !ROUTINE: MAIN_DO_LOOP
33  C     !INTERFACE:  C     !INTERFACE:
34        SUBROUTINE MAIN_DO_LOOP( myTime, myIter, myThid )        SUBROUTINE MAIN_DO_LOOP( iloop, myTime, myIter, myThid )
35    
36  C     !DESCRIPTION: \bv  C     !DESCRIPTION: \bv
37  C     *================================================================*  C     *================================================================*
# Line 224  CEOP Line 224  CEOP
224  c     >>>>>>>>>>>>>>>>>>>>>>>>>>>   LOOP   <<<<<<<<<<<<<<<<<<<<<<<<<<<<  c     >>>>>>>>>>>>>>>>>>>>>>>>>>>   LOOP   <<<<<<<<<<<<<<<<<<<<<<<<<<<<
225  c     >>>>>>>>>>>>>>>>>>>>>>>>>>>  STARTS  <<<<<<<<<<<<<<<<<<<<<<<<<<<<  c     >>>>>>>>>>>>>>>>>>>>>>>>>>>  STARTS  <<<<<<<<<<<<<<<<<<<<<<<<<<<<
226    
 c++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++  
 #ifndef ALLOW_AUTODIFF_OPENAD  
 c++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++  
 # ifdef ALLOW_AUTODIFF  
 #  ifdef ALLOW_TAMC_CHECKPOINTING  
   
       max_lev4=nTimeSteps/(nchklev_1*nchklev_2*nchklev_3)+1  
       max_lev3=nTimeSteps/(nchklev_1*nchklev_2)+1  
       max_lev2=nTimeSteps/nchklev_1+1  
   
 c**************************************  
 #   ifdef ALLOW_DIVIDED_ADJOINT  
 CADJ loop = divided  
 #   endif  
 c**************************************  
   
 #   ifdef AUTODIFF_4_LEVEL_CHECKPOINT  
       do ilev_4 = 1,nchklev_4  
          if(ilev_4.le.max_lev4) then  
 c**************************************  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 4 , 0, mythid )  
 #endif  
             CALL AUTODIFF_STORE( myThid )  
 #include "checkpoint_lev4_directives.h"  
             CALL AUTODIFF_RESTORE( myThid )  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 4 , 1, mythid )  
 #endif  
 c**************************************  
 c--     Initialise storage for the middle loop.  
 CADJ    INIT tapelev3 = USER  
 #   endif /* AUTODIFF_4_LEVEL_CHECKPOINT */  
   
 #   ifndef AUTODIFF_2_LEVEL_CHECKPOINT  
       do ilev_3 = 1,nchklev_3  
          if(ilev_3.le.max_lev3) then  
 c**************************************  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 3 , 0, mythid )  
 #endif  
             CALL AUTODIFF_STORE( myThid )  
 #include "checkpoint_lev3_directives.h"  
             CALL AUTODIFF_RESTORE( myThid )  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 3 , 1, mythid )  
 #endif  
 c**************************************  
 c--     Initialise storage for the middle loop.  
 CADJ    INIT tapelev2 = USER  
 #   endif /* AUTODIFF_2_LEVEL_CHECKPOINT */  
   
         do ilev_2 = 1,nchklev_2  
          if(ilev_2.le.max_lev2) then  
 c**************************************  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 2 , 0, mythid )  
 #endif  
             CALL AUTODIFF_STORE( myThid )  
 #include "checkpoint_lev2_directives.h"  
             CALL AUTODIFF_RESTORE( myThid )  
 #ifdef ALLOW_AUTODIFF_WHTAPEIO  
             CALL AUTODIFF_WHTAPEIO_SYNC( 2 , 1, mythid )  
 #endif  
 c**************************************  
   
 # endif /* ALLOW_TAMC_CHECKPOINTING */  
   
 c**************************************  
 c--  
 c--       Initialize storage for the innermost loop.  
 c--       Always check common block sizes for the checkpointing!  
 c--  
 CADJ INIT comlev1        = COMMON,nchklev_1  
 CADJ INIT comlev1_bibj   = COMMON,nchklev_1*nsx*nsy*nthreads_chkpt  
 CADJ INIT comlev1_bibj_k = COMMON,nchklev_1*nsx*nsy*nr*nthreads_chkpt  
 c--  
 #   ifdef ALLOW_KPP  
 CADJ INIT comlev1_kpp    = COMMON,nchklev_1*nsx*nsy  
 CADJ INIT comlev1_kpp_k  = COMMON,nchklev_1*nsx*nsy*nr  
 #   endif /* ALLOW_KPP */  
 c--  
 #   ifdef ALLOW_GMREDI  
 CADJ INIT comlev1_gmredi_k_gad  
 CADJ &    = COMMON,nchklev_1*nsx*nsy*nr*nthreads_chkpt*maxpass  
 #   endif /* ALLOW_GMREDI */  
 c--  
 #   ifdef ALLOW_PTRACERS  
 CADJ INIT comlev1_bibj_ptracers = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nthreads_chkpt*PTRACERS_num  
 CADJ INIT comlev1_bibj_k_ptracers = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nthreads_chkpt*PTRACERS_num*nr  
 #   endif /* ALLOW_PTRACERS */  
 c--  
 #   ifndef DISABLE_MULTIDIM_ADVECTION  
 CADJ INIT comlev1_bibj_gad = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nthreads_chkpt*maxpass  
 CADJ INIT comlev1_bibj_k_gad = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nr*nthreads_chkpt*maxpass  
 CADJ INIT comlev1_bibj_k_gad_pass = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nr*nthreads_chkpt*maxpass*maxpass  
 #   endif /* DISABLE_MULTIDIM_ADVECTION */  
 c--  
 #   ifdef ALLOW_MOM_COMMON  
 #   ifndef AUTODIFF_DISABLE_LEITH  
 CADJ INIT comlev1_mom_ijk_loop  
 CADJ &     = COMMON,nchklev_1*  
 CADJ &       (snx+2*olx)*nsx*(sny+2*oly)*nsy*nr*nthreads_chkpt  
 #   endif /* AUTODIFF_DISABLE_LEITH */  
 #   endif /* ALLOW_MOM_COMMON */  
 c--  
 #   if (defined (ALLOW_EXF) && defined (ALLOW_BULKFORMULAE))  
 CADJ INIT comlev1_exf_1  
 CADJ &     = COMMON,nchklev_1*snx*nsx*sny*nsy*nthreads_chkpt  
 CADJ INIT comlev1_exf_2  
 CADJ &     = COMMON,niter_bulk*nchklev_1*snx*nsx*sny*nsy*nthreads_chkpt  
 #   endif /* ALLOW_BULKFORMULAE */  
 c--  
 #   ifdef ALLOW_SEAICE  
 #    ifdef SEAICE_ALLOW_DYNAMICS  
 CADJ INIT comlev1_dynsol = COMMON,nchklev_1*MPSEUDOTIMESTEPS  
 #    ifdef SEAICE_LSR_ADJOINT_ITER  
 CADJ INIT comlev1_dyniter =  
 CADJ &    COMMON,nchklev_1*MPSEUDOTIMESTEPS*SOLV_MAX_FIXED  
 #    endif  
 #    endif  
 #   ifdef SEAICE_ALLOW_EVP  
 CADJ INIT comlev1_evp = COMMON,nEVPstepMax*nchklev_1  
 #   endif  
 #   ifdef SEAICE_MULTICATEGORY  
 CADJ INIT comlev1_multdim  
 CADJ &    = COMMON,nchklev_1*nsx*nsy*nthreads_chkpt*multdim  
 #   endif  
 #    ifndef DISABLE_MULTIDIM_ADVECTION  
 CADJ INIT comlev1_bibj_k_gadice = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nthreads_chkpt*maxpass  
 CADJ INIT comlev1_bibj_k_gadice_pass = COMMON,  
 CADJ &    nchklev_1*nsx*nsy*nthreads_chkpt*maxpass*maxpass  
 #    endif /* DISABLE_MULTIDIM_ADVECTION */  
 # endif /* ALLOW_SEAICE */  
 c--  
 #   ifdef ALLOW_THSICE  
 CADJ INIT comlev1_thsice_1  
 CADJ &     = COMMON,nchklev_1*snx*nsx*sny*nsy*nthreads_chkpt  
 CADJ INIT comlev1_thsice_2  
 CADJ &     = COMMON,nchklev_1*snx*nsx*sny*nsy*nlyr*nthreads_chkpt  
 CADJ INIT comlev1_thsice_3  
 CADJ &     = COMMON,nchklev_1*snx*nsx*sny*nsy*MaxTsf*nthreads_chkpt  
 CADJ INIT comlev1_thsice_4  
 CADJ &     = COMMON,nchklev_1*nsx*nsy*maxpass*nthreads_chkpt  
 CADJ INIT comlev1_thsice_s4t  
 CADJ &     = COMMON,nchklev_1*nsx*nsy*maxtsf*nthreads_chkpt  
 #   endif /* ALLOW_THSICE */  
 c--  
 #   ifdef ALLOW_STREAMICE  
 CADJ INIT comlev1_stream_nl = COMMON,nchklev_1*streamice_max_nl  
 CADJ INIT comlev1_stream_front = COMMON,nchklev_1*4  
 CADJ INIT comlev1_stream_ij  
 CADJ &     = COMMON,nchklev_1*4*(snx+2)*nsx*(sny+2)*nsy  
 CADJ INIT comlev1_stream_hybrid  
 CADJ &     = COMMON,nchklev_1*snx*nsx*sny*nsy*nr*nthreads_chkpt  
 #   endif  
 c--  
 #   ifdef ALLOW_CG2D_NSA  
 CADJ INIT comlev1_cg2d  
 CADJ &     = COMMON,nchklev_1*nthreads_chkpt  
 CADJ INIT comlev1_cg2d_iter  
 CADJ &     = COMMON,nchklev_1*nthreads_chkpt*numItersMax  
 #   endif  
 c--  
 c**************************************  
   
 #ifdef STORE_LOADEDREC_TEST  
           DO bj = myByLo(myThid), myByHi(myThid)  
            DO bi = myBxLo(myThid), myBxHi(myThid)  
             loadedRec(bi,bj) = 0  
            ENDDO  
           ENDDO  
 #endif /* STORE_LOADEDREC_TEST */  
   
 #ifdef ALLOW_TAMC_CHECKPOINTING  
   
           do ilev_1 = 1,nchklev_1  
   
 c--         The if-statement below introduces a some flexibility in the  
 c--         choice of the 3-tupel ( nchklev_1, nchklev_2, nchklev_3 ).  
   
             iloop = (ilev_2 - 1)*nchklev_1 + ilev_1  
 #    ifndef AUTODIFF_2_LEVEL_CHECKPOINT  
      &            + (ilev_3 - 1)*nchklev_2*nchklev_1  
 #    endif  
 #    ifdef AUTODIFF_4_LEVEL_CHECKPOINT  
      &            + (ilev_4 - 1)*nchklev_3*nchklev_2*nchklev_1  
 #    endif  
   
             if ( iloop .le. nTimeSteps ) then  
   
 #  else /* ALLOW_TAMC_CHECKPOINTING  undefined */  
   
       DO iloop = 1, nTimeSteps  
   
 #  endif /* ALLOW_TAMC_CHECKPOINTING */  
 # endif /* ALLOW_AUTODIFF */  
   
 #endif /* undef ALLOW_AUTODIFF_OPENAD */  
   
227  #ifdef ALLOW_AUTODIFF_OPENAD  #ifdef ALLOW_AUTODIFF_OPENAD
228        DO iloop = 1, nTimeSteps        DO iloop = 1, nTimeSteps
229  #endif  #endif
230    
 #ifndef ALLOW_AUTODIFF  
   
 c--   Start the main loop of adjoint_Objfunc. Automatic differentiation  
 c--   NOT enabled.  
       DO iloop = 1, nTimeSteps  
   
 #endif /* ALLOW_AUTODIFF */  
   
 c--     >>> Loop body start <<<  
   
 #ifdef ALLOW_AUTODIFF_TAMC  
         nIter0 = NINT( (startTime-baseTime)/deltaTClock )  
 # ifndef ALLOW_AUTODIFF_OPENAD  
         ikey_dynamics = ilev_1  
 # endif  
 #endif  
   
 #ifdef ALLOW_ECCO  
 #ifdef ALLOW_ECCO_EVOLUTION  
 #ifdef ALLOW_DEBUG  
       IF (debugMode) CALL DEBUG_CALL('cost_averagesfields',myThid)  
 #endif  
 c--   Accumulate time averages of temperature, salinity  
 #ifdef ALLOW_AUTODIFF  
 C--   Reset the model iteration counter and the model time.  
       myIter = nIter0 + (iloop-1)  
       myTime = startTime + float(iloop-1)*deltaTClock  
 #endif  
       CALL TIMER_START('COST_AVERAGESFIELDS [MAIN_DO_LOOP]',mythid)  
       CALL COST_AVERAGESFIELDS( mytime, mythid )  
       CALL TIMER_STOP ('COST_AVERAGESFIELDS [MAIN_DO_LOOP]',mythid)  
 #endif /* ALLOW_ECCO_EVOLUTION */  
 #endif /* ALLOW_ECCO */  
   
 #ifdef ALLOW_PROFILES  
       IF (usePROFILES) THEN  
 #ifdef ALLOW_DEBUG  
         IF (debugMode) CALL DEBUG_CALL('profiles_inloop',myThid)  
 #endif  
 c--     Accumulate in-situ time averages of theta, salt, and SSH.  
 #ifdef ALLOW_AUTODIFF  
 C--   Reset the model iteration counter and the model time.  
         myIter = nIter0 + (iloop-1)  
         myTime = startTime + float(iloop-1)*deltaTClock  
 #endif  
         CALL TIMER_START('PROFILES_INLOOP    [MAIN_DO_LOOP]', mythid)  
         CALL PROFILES_INLOOP( mytime, mythid )  
         CALL TIMER_STOP ('PROFILES_INLOOP    [MAIN_DO_LOOP]', mythid)  
       ENDIF  
 #endif  
   
231  #ifdef ALLOW_DEBUG  #ifdef ALLOW_DEBUG
232        IF (debugMode) CALL DEBUG_CALL('FORWARD_STEP',myThid)        IF (debugMode) CALL DEBUG_CALL('FORWARD_STEP',myThid)
233  #endif  #endif
# Line 499  C--   Reset the model iteration counter Line 242  C--   Reset the model iteration counter
242          CALL TIMER_STOP ('FORWARD_STEP        [MAIN_DO_LOOP]',mythid)          CALL TIMER_STOP ('FORWARD_STEP        [MAIN_DO_LOOP]',mythid)
243  #endif  #endif
244    
245  c--     >>> Loop body end <<<  #ifdef ALLOW_AUTODIFF_OPENAD
 #ifdef ALLOW_AUTODIFF  
 # ifndef ALLOW_AUTODIFF_OPENAD  
   
 #   ifdef ALLOW_TAMC_CHECKPOINTING  
             endif  
           enddo  
           endif  
         enddo  
 #    ifndef AUTODIFF_2_LEVEL_CHECKPOINT  
         endif  
       enddo  
 #    endif  
 #    ifdef AUTODIFF_4_LEVEL_CHECKPOINT  
        endif  
       enddo  
 #    endif  
 #   else /* ndef ALLOW_TAMC_CHECKPOINTING */  
       ENDDO  
 #   endif /* ALLOW_TAMC_CHECKPOINTING */  
 # else /* ndef ALLOW_AUTODIFF_OPENAD */  
       ENDDO  
 # endif /* ALLOW_AUTODIFF_OPENAD */  
 #else /* ALLOW_AUTODIFF */  
246        ENDDO        ENDDO
247  #endif /* ALLOW_AUTODIFF */  #endif
248    
249  #ifdef ALLOW_DEBUG  #ifdef ALLOW_DEBUG
250        IF (debugMode) CALL DEBUG_LEAVE('MAIN_DO_LOOP',myThid)        IF (debugMode) CALL DEBUG_LEAVE('MAIN_DO_LOOP',myThid)
Legend:
Removed from v.1.10  
changed lines
  Added in v.1.11
  ViewVC Help
Powered by ViewVC 1.1.22