Skip to content

Commit 2d79d23

Browse files
author
Ian Schweer
committed
Disable compile here based on CI result
1 parent a0b23cd commit 2d79d23

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytensor/link/pytorch/dispatch/scalar.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import torch
2+
import torch.compiler
23

34
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
45
from pytensor.scalar.basic import ScalarOp
@@ -68,4 +69,4 @@ def scalar_loop(*args):
6869
carry = update(*carry, *constants)
6970
return torch.stack(carry)
7071

71-
return scalar_loop
72+
return torch.compiler.disable(scalar_loop)

0 commit comments

Comments
 (0)