!--------------------------------------------------------------------------
   !---------------------------- theta ----------------------------------
   IF(caldyn_eta==eta_mass) THEN ! Compute mass
      ! compute mass_col from rhodz
      !$OMP DO SCHEDULE(STATIC)
      DO ij=1,primal_num
         mass_col(ij)=0.
         DO l = 1,llm
            mass_col(ij) = mass_col(ij) + rhodz(l,ij)
         END DO
      END DO
      !$OMP END DO
      !$OMP DO SCHEDULE(STATIC)
      DO ij = 1, primal_num
         !DIR$ SIMD
         DO l = 1, llm
            rhodz(l,ij) = MASS_DAK(l,ij) + mass_col(ij)*MASS_DBK(l,ij)
         END DO
      END DO
      !$OMP END DO
   END IF
   DO iq=1,nqdyn
      !$OMP DO SCHEDULE(STATIC)
      DO ij = 1, primal_num
         !DIR$ SIMD
         DO l = 1, llm
            theta(l,ij,iq) = theta_rhodz(l,ij,iq)/rhodz(l,ij)
         END DO
      END DO
      !$OMP END DO
   END DO
   !---------------------------- theta ----------------------------------
   !--------------------------------------------------------------------------