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 913ea6b commit 8286e1bCopy full SHA for 8286e1b
tests/link/pytorch/test_blockwise.py
@@ -5,11 +5,11 @@
5
import pytensor.tensor as pt
6
from pytensor.graph.basic import Apply
7
from pytensor.graph.op import Op
8
-from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
9
from pytensor.tensor.blockwise import Blockwise
10
11
12
torch = pytest.importorskip("torch")
+basic = pytest.importorskip("pytensor.link.pytorch.dispatch.basic")
13
14
15
class TestOp(Op):
@@ -27,7 +27,7 @@ def perform(self, *_):
27
raise RuntimeError("In perform")
28
29
30
-@pytorch_funcify.register(TestOp)
+@basic.pytorch_funcify.register(TestOp)
31
def evaluate_test_op(op, **_):
32
@torch.compiler.disable(recursive=False)
33
def func(a, b):
0 commit comments