-
Notifications
You must be signed in to change notification settings - Fork 38
docs: add tutorial on raising #1935
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.
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
base: main
Are you sure you want to change the base?
Conversation
02ecfaa to
7f2a3d2
Compare
| disable_auto_batching_passes=true | ||
| ) compute_attractive_force(positions_ra, masses_ra, 2.0f0) | ||
| ``` | ||
|
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
do we want to show the code here for example?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
you mean the manually written one with broadcasting? For the hlo, those will be printed out automatically
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ah awesome, yeah I just meant that it would be cool if we could see the before and after IR. If its automatically produced, even better!
|
needs EnzymeAD/Enzyme-JAX#1668 + corresponding jll |
docs/src/tutorials/raising.md
Outdated
|
|
||
| ## Raising GPU Kernels | ||
|
|
||
| <!-- TODO: write this section --> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I have written this and I can complete with a tutorial 5fe6e01
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yeah for ease @Pangoraw mind just pushing that commit to this branch here?
No description provided.