Skip to content

Commit e06994f

Browse files
author
Ian Schweer
committed
Recursive false for torch compiler
1 parent 623dfbe commit e06994f

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
@@ -98,4 +98,4 @@ def scalar_loop(steps, *start_and_constants):
9898
else:
9999
return carry
100100

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

0 commit comments

Comments
 (0)