The following changes/improvements for PBL/SATMEDMF are from @JongilHan66:

a) As the current large horizontal momentum diffusivity (Kh) at
large-scale and meso-scale (LSMS) looks too large, Kh is computed at zl
and modified as Kh=Kq=0.4*elmh*sqrt(tke),
elmh=min(1.5*sqrt(zlup*zldn,500), where zlup is parcel's upward travel
length and zldn is parcel's downward travel length with tke momentum.
b) Scale-aware non-local mass flux (xmf, xmfd) is now given as
xmf=(1-sigma)^2 * (sigma*wu) rather than the current xmf=pfnl*c1*wu,
where wu is updraft velocity, c1 is a constant, sigma is updraft
fraction, and pfnl is partition function for nonlocal flux. The constant
c1 is the updraft fraction in LSMS and it should be sigma in small
scales as sigma becomes larger than c1 in small scales. Also, the
scale-awareness parameter, (1-sigma)^2, is more physically based
following Arakawa and Wu's study than the empirical parameter, pfnl.
c) Refinement of scl (height of 50% of max tke): max tke can be
occasionally found in Jet stream and then scl would be too large. In the
update, we look for max tke below 2.5km height and then compute scl and
limit to 2.5km (i.e., scl=min(max(scl,500),2500))
d) ri(gradient Richardson number) is computed as ri=bf/shr2 (1dtke)
rather than ri=bf/(def_1+def_2), where bf is buoyancy, because the
parameters given as a function of ri (such as Prandtl number) is derived
based on ri=bf/shr2.
e) TTE-EDMF (total turbulent energy based EDMF scheme) is combined into
sa3dtke scheme as it shows slightly better GFS forecast skill than
tke-based scheme. The logical parameter to conduct TTE-EDMF is
'tte_edmf'.

Author: BinLiu-NOAA

physics/PBL/SATMEDMF/mfscuq.f

@@ -15,9 +15,7 @@ subroutine mfscuq(im,ix,km,kmscu,ntcw,ntrac1,delt,
      &   cnvflg,zl,zm,q1,t1,u1,v1,plyr,pix,
      &   thlx,thvx,thlvx,gdx,thetae,
      &   krad,mrad,radmin,buo,wush,tkemean,vez0fun,xmfd,
-     &   tcdo,qcdo,ucdo,vcdo,xlamdeq,a1,
-!The following flag is for SA-3D-TKE (kyf)
-     &   sa3dtke)
+     &   tcdo,qcdo,ucdo,vcdo,xlamdeq,a1)
 !
       use machine , only : kind_phys
       use funcphys , only : fpvs
@@ -33,7 +31,6 @@ subroutine mfscuq(im,ix,km,kmscu,ntcw,ntrac1,delt,
       integer   krad(im), mrad(im)
 !
       logical cnvflg(im)
-      logical sa3dtke !flag for SA-3D-TKE scheme (kyf)
       real(kind=kind_phys) delt
       real(kind=kind_phys) q1(ix,km,ntrac1),t1(ix,km),
      &                     u1(ix,km),      v1(ix,km),
@@ -430,16 +427,6 @@ subroutine mfscuq(im,ix,km,kmscu,ntcw,ntrac1,delt,
         endif
       enddo
 !
-!> - Set updraft fraction to 0 when using SA-3D-TKE scheme (kyf)
-!! Scale-aware capability is done with pfnl in satmedmfvdifq.F
-!! Zhu et al. (2025)
-!
-      if (sa3dtke) then
-        do i = 1, im
-          sigma(i) = 0.
-        enddo
-      endif
-!
 !> - Compute scale-aware function based on 
 !! Arakawa and Wu (2013) \cite arakawa_and_wu_2013
 !
@@ -460,6 +447,9 @@ subroutine mfscuq(im,ix,km,kmscu,ntcw,ntrac1,delt,
         do i = 1, im
           if(cnvflg(i) .and.
      &       (k >= mrad(i) .and. k < krad(i))) then
+             if (sigma(i) > ra1(i)) then
+               xmfd(i,k) = sigma(i) * xmfd(i,k) / ra1(i)
+             endif
              xmfd(i,k) = scaldfunc(i) * xmfd(i,k)
              dz   = zl(i,k+1) - zl(i,k)
              xmmx = dz / dt2