jax
45008fab - Include `ad_checkpoint` primitives in `roofline`.

Commit
251 days ago
Include `ad_checkpoint` primitives in `roofline`. These are registered with `register_standard_roofline` and thus contribute no costs to the roofline result, but registering the primitives prevents crashes when users attempt to use them. PiperOrigin-RevId: 775880161
Author
Parents
Loading