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):
863863 op = at .hyp2f1 ,
864864 expected = expected_hyp2f1 ,
865865 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 ,
867870)
868871
869872TestHyp2F1InplaceBroadcast = makeBroadcastTester (
@@ -874,6 +877,10 @@ def test_beta_inc_stan_grad_combined(self):
874877)
875878
876879
880+ @pytest .mark .skipif (
881+ config .mode == "FAST_COMPILE" ,
882+ reason = "Elemwise of Hyp2F1Grad fails in Python mode due to 32 operand limit" ,
883+ )
877884class TestHyp2F1Grad :
878885 few_iters_case = (
879886 2.0 ,
You can’t perform that action at this time.
0 commit comments