Skip to content

Commit 8355fcb

Browse files
author
Ian Schweer
committed
Disable compile here based on CI result
1 parent 4990e3a commit 8355fcb

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 (
@@ -80,4 +81,4 @@ def scalar_loop(*args):
8081
carry = update(*carry, *constants)
8182
return torch.stack(carry)
8283

83-
return scalar_loop
84+
return torch.compiler.disable(scalar_loop)

0 commit comments

Comments
 (0)