Skip to content

Commit 2844bc4

Browse files
author
Ian Schweer
committed
Disable compile here based on CI result
1 parent ae1c9da commit 2844bc4

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,6 +1,7 @@
11
import importlib
22

33
import torch
4+
import torch.compiler
45

56
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
67
from pytensor.scalar.basic import (
@@ -92,4 +93,4 @@ def scalar_loop(*args):
9293
carry = update(*carry, *constants)
9394
return torch.stack(carry)
9495

95-
return scalar_loop
96+
return torch.compiler.disable(scalar_loop)

0 commit comments

Comments
 (0)