Merge pull request #1368 from stan-dev/binomial-opencl

Mark binomial logit and glm as opencl-supported
stan-dev · Oct 2, 2024 · 41f4d0c · 41f4d0c
2 parents c5721d7 + b4285d0
commit 41f4d0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/stan_math_backend/Transform_Mir.ml b/src/stan_math_backend/Transform_Mir.ml
@@ -166,7 +166,8 @@ let opencl_supported_functions =
   ; "ordered_logistic_glm_lpmf"; "pareto_lpdf"; "pareto_type_2_lpdf"
   ; "poisson_lpmf"; "poisson_log_lpmf"; "poisson_log_glm_lpmf"; "rayleigh_lpdf"
   ; "scaled_inv_chi_square_lpdf"; "skew_normal_lpdf"; "std_normal_lpdf"
-  ; "student_t_lpdf"; "uniform_lpdf"; "weibull_lpdf" ]
+  ; "student_t_lpdf"; "uniform_lpdf"; "weibull_lpdf"; "binomial_logit_lpmf"
+  ; "binomial_logit_glm_lpmf" ]
   |> String.Set.of_list
 
 let opencl_suffix = "_opencl__"