Skip to content

Commit 2f94d1a

Browse files
committed
Fix wrong backend in Numba logsumexp benchmark
1 parent c111364 commit 2f94d1a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/link/numba/test_elemwise.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -548,7 +548,7 @@ def test_logsumexp_benchmark(size, axis, benchmark):
548548
rng = np.random.default_rng(23920)
549549
X_val = rng.normal(size=size)
550550

551-
X_lse_fn = pytensor.function([X], X_lse, mode="JAX")
551+
X_lse_fn = pytensor.function([X], X_lse, mode="NUMBA")
552552

553553
# JIT compile first
554554
_ = X_lse_fn(X_val)

0 commit comments

Comments
 (0)