Skip to content

Commit cb7e4db

Browse files
author
Ian Schweer
committed
Recursive false for torch compiler
1 parent ad04464 commit cb7e4db

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
@@ -74,4 +74,4 @@ def scalar_loop(steps, *start_and_constants):
7474
else:
7575
return carry
7676

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

0 commit comments

Comments
 (0)