Skip to content

Conversation

@avik-pal
Copy link
Collaborator

@avik-pal avik-pal commented Dec 6, 2025

No description provided.

@avik-pal avik-pal force-pushed the ap/stablehlo_raising branch from 02ecfaa to 7f2a3d2 Compare December 6, 2025 18:06
@avik-pal avik-pal changed the title docs: add tutorial on raising loops [skip ci] docs: add tutorial on raising Dec 6, 2025
disable_auto_batching_passes=true
) compute_attractive_force(positions_ra, masses_ra, 2.0f0)
```

Copy link
Member

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?

Copy link
Collaborator Author

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

Copy link
Member

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!

@avik-pal
Copy link
Collaborator Author

avik-pal commented Dec 7, 2025

needs EnzymeAD/Enzyme-JAX#1668 + corresponding jll

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants