Skip to content

Commit d65d501

Browse files
author
Ian Schweer
committed
Recursive false for torch compiler
1 parent d45fb5e commit d65d501

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)