expose bernoulli_logit_glm_rng

Author: adamhaber

src/middle/Stan_math_signatures.ml

@@ -961,6 +961,26 @@ let () =
     , ReturnType UReal
     , [UArray UInt; URowVector; UVector; UVector]
     , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [UMatrix; UReal; UVector]
+    , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [UMatrix; UVector; UVector]
+    , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [URowVector; UReal; UVector]
+    , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [URowVector; UVector; UVector]
+    , AoS ) ;
   add_binary_vec_int_real "bessel_first_kind" AoS ;
   add_binary_vec_int_real "bessel_second_kind" AoS ;
   add_binary_vec "beta" AoS ;

test/integration/good/function-signatures/distributions/univariate/discrete/bernoulli/bernoulli_logit_glm.stan

@@ -11,7 +11,6 @@ transformed data {
   real transformed_data_real;
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha_v, d_beta_v);
-  transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y| d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y| d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_rv, d_alpha, d_beta_v);
@@ -37,4 +36,11 @@ transformed parameters {
 model {
   y_p ~ normal(0, 1);
 }
+generated quantities {
+  int y_gen_ar[N];
+  y_gen_ar = bernoulli_logit_glm_rng(d_x_m, d_alpha, d_beta_v);
+  y_gen_ar = bernoulli_logit_glm_rng(d_x_m, d_alpha_v, d_beta_v);
+  y_gen_ar = bernoulli_logit_glm_rng(d_x_rv, d_alpha, d_beta_v);
+  y_gen_ar = bernoulli_logit_glm_rng(d_x_rv, d_alpha_v, d_beta_v);
+}
 

test/integration/signatures/stan_math_sigs.expected

@@ -463,6 +463,10 @@ bernoulli_logit_glm_lpmf(array[] int, row_vector, real, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, row_vector, vector, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, matrix, real, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, matrix, vector, vector) => real
+bernoulli_logit_glm_rng(row_vector, real, vector) => array[] int
+bernoulli_logit_glm_rng(row_vector, vector, vector) => array[] int
+bernoulli_logit_glm_rng(matrix, real, vector) => array[] int
+bernoulli_logit_glm_rng(matrix, vector, vector) => array[] int
 bernoulli_logit_log(int, real) => real
 bernoulli_logit_log(int, vector) => real
 bernoulli_logit_log(int, row_vector) => real