jeertmans/DiffeRT
[DOC] Document when `compute_paths` can be JIT-compiled
Ouverte
#337 ouverte le 23 oct. 2025
documentationgood first issueneeds-triage
Métriques du dépôt
- Stars
- (64 étoiles)
- Métriques de merge PR
- (Merge moyen 2j 7h) (24 PRs mergées en 30 j)
Description
Terms
- Checked the existing issues and discussions to see if my issue had not already been reported;
Describe the issue
As discussed in https://github.com/jeertmans/DiffeRT/pull/326#issuecomment-3437022985, the compute_paths method exhibits some challenges when it comes to JIT-compilation, as some cases, e.g., when setting disconnect_inactive_triangles to True, will create output arrays whose shapes depend on the values of the input arrays. This is not compatible with JAX's tracing system, this should probably be documented somewhere.
Affected page(s)
- https://differt.eertmans.be/latest/reference/_autosummary/differt.scene.TriangleScene.html#differt.scene.TriangleScene.compute_paths
- https://differt.eertmans.be/latest/notebooks/performance_tips.html
Issue type
Documentation enhancement
Recommended fix or suggestions
No response