@@ -32,7 +32,8 @@ namespace internal {
3232/* *
3333 * Computes log q and d(log q) / d(alpha) using continued fraction.
3434 */
35- template <bool any_fvar, bool partials_fvar, typename T_shape, typename T1, typename T2>
35+ template <bool any_fvar, bool partials_fvar, typename T_shape, typename T1,
36+ typename T2>
3637inline std::optional<std::pair<return_type_t <T1, T2>, return_type_t <T1, T2>>>
3738eval_q_cf (const T1& alpha, const T2& beta_y) {
3839 using scalar_t = return_type_t <T1, T2>;
@@ -89,8 +90,7 @@ eval_q_log1m(const T1& alpha, const T2& beta_y) {
8990 auto log_Q_fvar = log1m (gamma_p (alpha_unit, beta_unit));
9091 out.second = log_Q_fvar.d_ ;
9192 } else {
92- out.second
93- = -grad_reg_lower_inc_gamma (alpha, beta_y) / exp (out.first );
93+ out.second = -grad_reg_lower_inc_gamma (alpha, beta_y) / exp (out.first );
9494 }
9595 }
9696 return std::optional{out};
@@ -128,9 +128,9 @@ inline return_type_t<T_y, T_shape, T_inv_scale> gamma_lccdf(
128128 scalar_seq_view<T_beta_ref> beta_vec (beta_ref);
129129 const size_t N = max_size (y, alpha, beta);
130130
131- constexpr bool any_fvar = is_fvar_v< scalar_type_t <T_y>>
132- || is_fvar_v<scalar_type_t <T_shape>>
133- || is_fvar_v<scalar_type_t <T_inv_scale>>;
131+ constexpr bool any_fvar
132+ = is_fvar_v<scalar_type_t <
133+ T_y>> || is_fvar_v< scalar_type_t <T_shape>> || is_fvar_v<scalar_type_t <T_inv_scale>>;
134134 constexpr bool partials_fvar = is_fvar_v<T_partials_return>;
135135
136136 for (size_t n = 0 ; n < N; n++) {
@@ -153,12 +153,15 @@ inline return_type_t<T_y, T_shape, T_inv_scale> gamma_lccdf(
153153 }
154154 std::optional<std::pair<T_partials_return, T_partials_return>> result;
155155 if (beta_y > alpha_val + 1.0 ) {
156- result = internal::eval_q_cf<any_fvar, partials_fvar, T_shape>(alpha_val, beta_y);
156+ result = internal::eval_q_cf<any_fvar, partials_fvar, T_shape>(alpha_val,
157+ beta_y);
157158 } else {
158- result = internal::eval_q_log1m<partials_fvar, T_shape>(alpha_val, beta_y);
159+ result
160+ = internal::eval_q_log1m<partials_fvar, T_shape>(alpha_val, beta_y);
159161 if (!result && beta_y > 0.0 ) {
160162 // Fallback to continued fraction if log1m fails
161- result = internal::eval_q_cf<any_fvar, partials_fvar, T_shape>(alpha_val, beta_y);
163+ result = internal::eval_q_cf<any_fvar, partials_fvar, T_shape>(
164+ alpha_val, beta_y);
162165 }
163166 }
164167 if (unlikely (!result)) {
0 commit comments