提交 2f94d1a8 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Fix wrong backend in Numba logsumexp benchmark

上级 c111364f
......@@ -548,7 +548,7 @@ def test_logsumexp_benchmark(size, axis, benchmark):
rng = np.random.default_rng(23920)
X_val = rng.normal(size=size)
X_lse_fn = pytensor.function([X], X_lse, mode="JAX")
X_lse_fn = pytensor.function([X], X_lse, mode="NUMBA")
# JIT compile first
_ = X_lse_fn(X_val)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论