Skip to content

Commit ed71acb

Browse files
author
Ian Schweer
committed
Recursive false for torch compiler
1 parent 960488f commit ed71acb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/link/pytorch/dispatch/scalar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,4 +86,4 @@ def scalar_loop(steps, *start_and_constants):
8686
else:
8787
return carry
8888

89-
return torch.compiler.disable(scalar_loop)
89+
return torch.compiler.disable(scalar_loop, recursive=False)

0 commit comments

Comments
 (0)