Skip to content

Commit f7349d0

Browse files
SteveBronderSteveBronder
authored
Merge pull request #1095 from SteveBronder/feature/aos-glms
update glm signatures to accept SoA matrices
2 parents 4f899ea + bd678a5 commit f7349d0

1 file changed

Lines changed: 35 additions & 35 deletions

File tree

‎src/middle/Stan_math_signatures.ml‎

Lines changed: 35 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -212,12 +212,12 @@ let distributions =
212212
; ([Lpdf; Ccdf; Cdf], "beta_proportion", [DVReal; DVReal; DIntAndReals], SoA)
213213
; (full_lpmf, "bernoulli", [DVInt; DVReal], SoA)
214214
; ([Lpmf; Rng], "bernoulli_logit", [DVInt; DVReal], SoA)
215-
; ([Lpmf], "bernoulli_logit_glm", [DVInt; DMatrix; DReal; DVector], AoS)
215+
; ([Lpmf], "bernoulli_logit_glm", [DVInt; DMatrix; DReal; DVector], SoA)
216216
; (full_lpmf, "binomial", [DVInt; DVInt; DVReal], SoA)
217217
; ([Lpmf], "binomial_logit", [DVInt; DVInt; DVReal], SoA)
218218
; ([Lpmf], "categorical", [DVInt; DVector], AoS)
219219
; ([Lpmf], "categorical_logit", [DVInt; DVector], AoS)
220-
; ([Lpmf], "categorical_logit_glm", [DVInt; DMatrix; DVector; DMatrix], AoS)
220+
; ([Lpmf], "categorical_logit_glm", [DVInt; DMatrix; DVector; DMatrix], SoA)
221221
; (full_lpdf, "cauchy", [DVReal; DVReal; DVReal], SoA)
222222
; (full_lpdf, "chi_square", [DVReal; DVReal], SoA)
223223
; ([Lpdf], "dirichlet", [DVectors; DVectors], SoA)
@@ -254,16 +254,16 @@ let distributions =
254254
; ( [Lpmf]
255255
, "neg_binomial_2_log_glm"
256256
, [DVInt; DMatrix; DReal; DVector; DReal]
257-
, AoS ); (full_lpdf, "normal", [DVReal; DVReal; DVReal], SoA)
258-
; ([Lpdf], "normal_id_glm", [DVector; DMatrix; DReal; DVector; DReal], AoS)
257+
, SoA ); (full_lpdf, "normal", [DVReal; DVReal; DVReal], SoA)
258+
; ([Lpdf], "normal_id_glm", [DVector; DMatrix; DReal; DVector; DReal], SoA)
259259
; ([Lpmf], "ordered_logistic", [DInt; DReal; DVector], SoA)
260-
; ([Lpmf], "ordered_logistic_glm", [DVInt; DMatrix; DVector; DVector], AoS)
260+
; ([Lpmf], "ordered_logistic_glm", [DVInt; DMatrix; DVector; DVector], SoA)
261261
; ([Lpmf], "ordered_probit", [DInt; DReal; DVector], SoA)
262262
; (full_lpdf, "pareto", [DVReal; DVReal; DVReal], SoA)
263263
; (full_lpdf, "pareto_type_2", [DVReal; DVReal; DVReal; DVReal], SoA)
264264
; (full_lpmf, "poisson", [DVInt; DVReal], SoA)
265265
; ([Lpmf; Rng], "poisson_log", [DVInt; DVReal], SoA)
266-
; ([Lpmf], "poisson_log_glm", [DVInt; DMatrix; DReal; DVector], AoS)
266+
; ([Lpmf], "poisson_log_glm", [DVInt; DMatrix; DReal; DVector], SoA)
267267
; (full_lpdf, "rayleigh", [DVReal; DVReal], SoA)
268268
; (full_lpdf, "scaled_inv_chi_square", [DVReal; DVReal; DVReal], SoA)
269269
; (full_lpdf, "skew_normal", [DVReal; DVReal; DVReal; DVReal], SoA)
@@ -936,22 +936,22 @@ let () =
936936
( "bernoulli_logit_glm_lpmf"
937937
, ReturnType UReal
938938
, [UArray UInt; UMatrix; UVector; UVector]
939-
, AoS ) ;
939+
, SoA ) ;
940940
add_unqualified
941941
( "bernoulli_logit_glm_lpmf"
942942
, ReturnType UReal
943943
, [UInt; UMatrix; UVector; UVector]
944-
, AoS ) ;
944+
, SoA ) ;
945945
add_unqualified
946946
( "bernoulli_logit_glm_lpmf"
947947
, ReturnType UReal
948948
, [UArray UInt; URowVector; UReal; UVector]
949-
, AoS ) ;
949+
, SoA ) ;
950950
add_unqualified
951951
( "bernoulli_logit_glm_lpmf"
952952
, ReturnType UReal
953953
, [UArray UInt; URowVector; UVector; UVector]
954-
, AoS ) ;
954+
, SoA ) ;
955955
add_unqualified
956956
( "bernoulli_logit_glm_rng"
957957
, ReturnType (UArray UInt)
@@ -983,12 +983,12 @@ let () =
983983
( "categorical_logit_glm_lpmf"
984984
, ReturnType UReal
985985
, [UArray UInt; URowVector; UVector; UMatrix]
986-
, AoS ) ;
986+
, SoA ) ;
987987
add_unqualified
988988
( "categorical_logit_glm_lpmf"
989989
, ReturnType UReal
990990
, [UInt; URowVector; UVector; UMatrix]
991-
, AoS ) ;
991+
, SoA ) ;
992992
add_unqualified ("append_col", ReturnType UMatrix, [UMatrix; UMatrix], AoS) ;
993993
add_unqualified ("append_col", ReturnType UMatrix, [UVector; UMatrix], AoS) ;
994994
add_unqualified ("append_col", ReturnType UMatrix, [UMatrix; UVector], AoS) ;
@@ -1666,69 +1666,69 @@ let () =
16661666
( "neg_binomial_2_log_glm_lpmf"
16671667
, ReturnType UReal
16681668
, [UArray UInt; UMatrix; UVector; UVector; UReal]
1669-
, AoS ) ;
1669+
, SoA ) ;
16701670
add_unqualified
16711671
( "neg_binomial_2_log_glm_lpmf"
16721672
, ReturnType UReal
16731673
, [UInt; UMatrix; UVector; UVector; UReal]
1674-
, AoS ) ;
1674+
, SoA ) ;
16751675
add_unqualified
16761676
( "neg_binomial_2_log_glm_lpmf"
16771677
, ReturnType UReal
16781678
, [UArray UInt; URowVector; UReal; UVector; UReal]
1679-
, AoS ) ;
1679+
, SoA ) ;
16801680
add_unqualified
16811681
( "neg_binomial_2_log_glm_lpmf"
16821682
, ReturnType UReal
16831683
, [UArray UInt; URowVector; UVector; UVector; UReal]
1684-
, AoS ) ;
1684+
, SoA ) ;
16851685
add_nullary "negative_infinity" ;
16861686
add_unqualified ("norm", ReturnType UReal, [UComplex], AoS) ;
16871687
add_unqualified
16881688
( "normal_id_glm_lpdf"
16891689
, ReturnType UReal
16901690
, [UVector; UMatrix; UVector; UVector; UReal]
1691-
, AoS ) ;
1691+
, SoA ) ;
16921692
add_unqualified
16931693
( "normal_id_glm_lpdf"
16941694
, ReturnType UReal
16951695
, [UReal; UMatrix; UReal; UVector; UReal]
1696-
, AoS ) ;
1696+
, SoA ) ;
16971697
add_unqualified
16981698
( "normal_id_glm_lpdf"
16991699
, ReturnType UReal
17001700
, [UReal; UMatrix; UVector; UVector; UReal]
1701-
, AoS ) ;
1701+
, SoA ) ;
17021702
add_unqualified
17031703
( "normal_id_glm_lpdf"
17041704
, ReturnType UReal
17051705
, [UReal; UMatrix; UReal; UVector; UVector]
1706-
, AoS ) ;
1706+
, SoA ) ;
17071707
add_unqualified
17081708
( "normal_id_glm_lpdf"
17091709
, ReturnType UReal
17101710
, [UReal; UMatrix; UVector; UVector; UVector]
1711-
, AoS ) ;
1711+
, SoA ) ;
17121712
add_unqualified
17131713
( "normal_id_glm_lpdf"
17141714
, ReturnType UReal
17151715
, [UVector; URowVector; UReal; UVector; UVector]
1716-
, AoS ) ;
1716+
, SoA ) ;
17171717
add_unqualified
17181718
( "normal_id_glm_lpdf"
17191719
, ReturnType UReal
17201720
, [UVector; URowVector; UVector; UVector; UReal]
1721-
, AoS ) ;
1721+
, SoA ) ;
17221722
add_unqualified
17231723
( "normal_id_glm_lpdf"
17241724
, ReturnType UReal
17251725
, [UVector; URowVector; UVector; UVector; UVector]
1726-
, AoS ) ;
1726+
, SoA ) ;
17271727
add_unqualified
17281728
( "normal_id_glm_lpdf"
17291729
, ReturnType UReal
17301730
, [UVector; URowVector; UReal; UVector; UReal]
1731-
, AoS ) ;
1731+
, SoA ) ;
17321732
add_nullary "not_a_number" ;
17331733
add_unqualified ("num_elements", ReturnType UInt, [UMatrix], SoA) ;
17341734
add_unqualified ("num_elements", ReturnType UInt, [UVector], SoA) ;
@@ -1755,32 +1755,32 @@ let () =
17551755
( "ordered_logistic_glm_lpmf"
17561756
, ReturnType UReal
17571757
, [UArray UInt; URowVector; UVector; UVector]
1758-
, AoS ) ;
1758+
, SoA ) ;
17591759
add_unqualified
17601760
( "ordered_logistic_glm_lpmf"
17611761
, ReturnType UReal
17621762
, [UInt; URowVector; UVector; UVector]
1763-
, AoS ) ;
1763+
, SoA ) ;
17641764
add_unqualified
17651765
( "ordered_logistic_log"
17661766
, ReturnType UReal
17671767
, [UArray UInt; UVector; UVector]
1768-
, AoS ) ;
1768+
, SoA ) ;
17691769
add_unqualified
17701770
( "ordered_logistic_log"
17711771
, ReturnType UReal
17721772
, [UArray UInt; UVector; UArray UVector]
1773-
, AoS ) ;
1773+
, SoA ) ;
17741774
add_unqualified
17751775
( "ordered_logistic_lpmf"
17761776
, ReturnType UReal
17771777
, [UArray UInt; UVector; UVector]
1778-
, AoS ) ;
1778+
, SoA ) ;
17791779
add_unqualified
17801780
( "ordered_logistic_lpmf"
17811781
, ReturnType UReal
17821782
, [UArray UInt; UVector; UArray UVector]
1783-
, AoS ) ;
1783+
, SoA ) ;
17841784
add_unqualified
17851785
("ordered_logistic_rng", ReturnType UInt, [UReal; UVector], AoS) ;
17861786
add_unqualified
@@ -1823,22 +1823,22 @@ let () =
18231823
( "poisson_log_glm_lpmf"
18241824
, ReturnType UReal
18251825
, [UArray UInt; UMatrix; UVector; UVector]
1826-
, AoS ) ;
1826+
, SoA ) ;
18271827
add_unqualified
18281828
( "poisson_log_glm_lpmf"
18291829
, ReturnType UReal
18301830
, [UInt; UMatrix; UVector; UVector]
1831-
, AoS ) ;
1831+
, SoA ) ;
18321832
add_unqualified
18331833
( "poisson_log_glm_lpmf"
18341834
, ReturnType UReal
18351835
, [UArray UInt; URowVector; UReal; UVector]
1836-
, AoS ) ;
1836+
, SoA ) ;
18371837
add_unqualified
18381838
( "poisson_log_glm_lpmf"
18391839
, ReturnType UReal
18401840
, [UArray UInt; URowVector; UVector; UVector]
1841-
, AoS ) ;
1841+
, SoA ) ;
18421842
add_unqualified ("polar", ReturnType UComplex, [UReal; UReal], AoS) ;
18431843
add_nullary "positive_infinity" ;
18441844
add_binary_vec "pow" AoS ;

0 commit comments

Comments
 (0)