File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -863,7 +863,10 @@ def test_beta_inc_stan_grad_combined(self):
863
863
op = at .hyp2f1 ,
864
864
expected = expected_hyp2f1 ,
865
865
good = _good_broadcast_quaternary_hyp2f1 ,
866
- grad = _good_broadcast_quaternary_hyp2f1 ,
866
+ # FIXME: FAST_COMPILE disables C implementation, and the
867
+ # Elemwise of the fused Hyp2F1 grad loop has more than 32 operands
868
+ # which is not supported by the python Elemwise perform
869
+ grad = _good_broadcast_quaternary_hyp2f1 if config .mode != "FAST_COMPILE" else None ,
867
870
)
868
871
869
872
TestHyp2F1InplaceBroadcast = makeBroadcastTester (
@@ -874,6 +877,10 @@ def test_beta_inc_stan_grad_combined(self):
874
877
)
875
878
876
879
880
+ @pytest .mark .skipif (
881
+ config .mode == "FAST_COMPILE" ,
882
+ reason = "Elemwise of Hyp2F1Grad fails in Python mode due to 32 operand limit" ,
883
+ )
877
884
class TestHyp2F1Grad :
878
885
few_iters_case = (
879
886
2.0 ,
You can’t perform that action at this time.
0 commit comments