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 a0b23cd commit 2d79d23Copy full SHA for 2d79d23
pytensor/link/pytorch/dispatch/scalar.py
@@ -1,4 +1,5 @@
1
import torch
2
+import torch.compiler
3
4
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
5
from pytensor.scalar.basic import ScalarOp
@@ -68,4 +69,4 @@ def scalar_loop(*args):
68
69
carry = update(*carry, *constants)
70
return torch.stack(carry)
71
- return scalar_loop
72
+ return torch.compiler.disable(scalar_loop)
0 commit comments