Skip to content

Commit 6ac04d7

Browse files
committed
Skip Hyp2F1Grad tests in FAST_COMPILE mode
1 parent 54b843d commit 6ac04d7

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/tensor/test_math_scipy.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff 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

869872
TestHyp2F1InplaceBroadcast = 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+
)
877884
class TestHyp2F1Grad:
878885
few_iters_case = (
879886
2.0,

0 commit comments

Comments
 (0)