提交 5b77cfba authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Temporarily remove heaviest param of jax test_logsumexp_benchmark

上级 6132203f
......@@ -109,7 +109,7 @@ def test_softmax_grad(axis):
compare_jax_and_py(fgraph, [get_test_value(i) for i in fgraph.inputs])
@pytest.mark.parametrize("size", [(10, 10), (1000, 1000), (10000, 10000)])
@pytest.mark.parametrize("size", [(10, 10), (1000, 1000)])
@pytest.mark.parametrize("axis", [0, 1])
def test_logsumexp_benchmark(size, axis, benchmark):
X = pt.matrix("X")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论