Slightly adjusted poly2 function.

hfp · Jun 6, 2024 · 09aa361 · 09aa361
1 parent e3475d0
commit 09aa361
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/hfx_load_balance_methods.F b/src/hfx_load_balance_methods.F
@@ -1810,10 +1810,12 @@ END FUNCTION estimate_basic
 !> \return ...
 ! **************************************************************************************************
       REAL(KIND=dp) FUNCTION poly2(x, a0, a1, a2)
-         INTEGER                                            :: x
-         REAL(KIND=dp)                                      :: a0, a1, a2
+         INTEGER, INTENT(IN)                                :: x
+         REAL(KIND=dp), INTENT(IN)                          :: a0, a1, a2
+         REAL(KIND=dp)                                      :: r
 
-         poly2 = a0 + a1*x + a2*x*x
+         r = REAL(x, KIND=dp)
+         poly2 = a0 + (a1 + a2*r)*r
       END FUNCTION poly2
 
    END FUNCTION cost_model