From f6d4a63118f535480def9319661afc3c163ae27a Mon Sep 17 00:00:00 2001
From: Jian Sun <sunjian@ucar.edu>
Date: Fri, 22 Mar 2024 11:41:37 -0600
Subject: [PATCH] fix the gpu runtime error on nvidia a100 with nvhpc/24.3

---
 rrtmgp-frontend/mo_gas_optics_rrtmgp.F90 | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/rrtmgp-frontend/mo_gas_optics_rrtmgp.F90 b/rrtmgp-frontend/mo_gas_optics_rrtmgp.F90
index c83601586..906c7cccb 100644
--- a/rrtmgp-frontend/mo_gas_optics_rrtmgp.F90
+++ b/rrtmgp-frontend/mo_gas_optics_rrtmgp.F90
@@ -801,6 +801,7 @@ function set_tsi(this, tsi) result(error_msg)
     character(len=128)                         :: error_msg !! Empty if successful 
 
     real(wp) :: norm
+    integer  :: igpt, length 
     ! ----------------------------------------------------------
     error_msg = ""
     if(tsi < 0._wp) then
@@ -809,12 +810,21 @@ function set_tsi(this, tsi) result(error_msg)
       !
       ! Scale the solar source function to the input tsi
       !
-      !$acc kernels
-      !$omp target
-      norm = 1._wp/sum(this%solar_source(:))
-      this%solar_source(:) = this%solar_source(:) * tsi * norm
-      !$acc end kernels
-      !$omp end target
+      norm = 0._wp
+      length = size(this%solar_source)
+      !$acc parallel loop gang vector reduction(+:norm)
+      !$omp target teams distribute parallel do simd reduction(+:norm)
+      do igpt = 1, length
+         norm = norm + this%solar_source(igpt)
+      end do
+
+      norm = 1._wp/norm
+
+      !$acc parallel loop gang vector
+      !$omp target teams distribute parallel do simd
+      do igpt = 1, length
+         this%solar_source(igpt) = this%solar_source(igpt) * tsi * norm
+      end do
     end if
 
   end function set_tsi