jeertmans/DiffeRT
[DOC] Document when `compute_paths` can be JIT-compiled
Aberta
#337 aberto em 23 de out. de 2025
documentationgood first issueneeds-triage
Métricas do repositório
- Stars
- (64 estrelas)
- Métricas de merge de PR
- (Mesclagem média 2d 7h) (24 fundiu PRs em 30d)
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