We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 960488f commit ed71acbCopy full SHA for ed71acb
pytensor/link/pytorch/dispatch/scalar.py
@@ -86,4 +86,4 @@ def scalar_loop(steps, *start_and_constants):
86
else:
87
return carry
88
89
- return torch.compiler.disable(scalar_loop)
+ return torch.compiler.disable(scalar_loop, recursive=False)
0 commit comments