We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e1fed99 commit a46adc8Copy full SHA for a46adc8
tests/link/pytorch/test_blas.py
@@ -31,6 +31,6 @@ def test_pytorch_BatchedDot():
31
inputs = [get_test_value(a)[:-1], get_test_value(b)]
32
opts = RewriteDatabaseQuery(include=[None], exclude=["cxx_only", "BlasOpt"])
33
pytorch_mode = Mode(PytorchLinker(), opts)
34
- pytensor_jax_fn = function(fgraph.inputs, fgraph.outputs, mode=pytorch_mode)
+ pytensor_pytorch_fn = function(fgraph.inputs, fgraph.outputs, mode=pytorch_mode)
35
with pytest.raises(TypeError):
36
- pytensor_jax_fn(*inputs)
+ pytensor_pytorch_fn(*inputs)
0 commit comments