Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Why is autodiff_thunk useful in forward mode? #1335

Closed
gdalle opened this issue Mar 7, 2024 · 1 comment
Closed

Why is autodiff_thunk useful in forward mode? #1335

gdalle opened this issue Mar 7, 2024 · 1 comment

Comments

@gdalle
Copy link
Contributor

gdalle commented Mar 7, 2024 •
edited
Loading

I understand the purpose of autodiff_thunk in reverse mode: it allows the creation of a tape so that subsequent pullbacks are evaluated much faster.

What about autodiff_thunk in forward mode though? Does it speed things up?
@gdalle gdalle mentioned this issue Mar 7, 2024
Closed
6 tasks
@gdalle gdalle changed the title Why is autodiff_deferred_thunk useful in forward mode? Why is autodiff_thunk useful in forward mode? Mar 8, 2024
This was referenced Mar 19, 2024
Closed
Open
@wsmoses
Copy link
Member

wsmoses commented May 13, 2024

If the original code is non-inferrable it may. It also enforces compilation for the specified types, if they differ from the runtime types of the arguments provided.

@wsmoses wsmoses closed this as completed May 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wsmoses @gdalle