jeertmans/DiffeRT
[DOC] Document when `compute_paths` can be JIT-compiled
オープン
#337 opened on 2025/10/23
documentationgood first issueneeds-triage
Repository metrics
- Stars
- (64 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
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