Ethanfel 83e4b5dd98 perf: add torch.compile to PyTorch fallback kernels ...

Wraps _pytorch_softsplat and _pytorch_costvol with torch.compile
for ~6x speedup on ROCm/non-cupy setups. Falls back to eager
execution gracefully if compilation fails.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

2026-04-11 10:13:30 +02:00

..

__init__.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

backbone.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

feature_extractor.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

flow_estimation.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

geometry.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

gmflow.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

matching.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

position.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

refine.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

softsplat.py

perf: add torch.compile to PyTorch fallback kernels

2026-04-11 10:13:30 +02:00

transformer.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

trident_conv.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

utils.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00

warplayer.py

Add SGM-VFI (CVPR 2024) frame interpolation support

2026-02-12 23:02:48 +01:00