Try using StaticOneTo instead of Val in unrolled_reduce

src/parameterized_tendencies/gravity_wave_drag/non_orographic_gravity_wave.jl

@@ -567,7 +567,7 @@ function waveforcing_column_accumulate!(
             # check break condition for each gravity waves and calculate momentum flux of breaking gravity waves at each level
             # We use the unrolled_reduce function here because it performs better for parallel execution on the GPU, avoiding type instabilities.
             (mask, fm) =
-                unrolled_reduce(Val(nc), (mask, FT1(0.0))) do (mask, fm), (n)
+                unrolled_reduce(StaticOneTo(nc), (mask, FT1(0.0))) do (mask, fm), (n)
                     if (mask[n]) == true
                         c_hat = c[n] - u_kp1 # c0mu
                         # f phase speed matches the wind speed, remove c(n) from the set of propagating waves.