kernel_utils/__init__.py
| 1 | """FlashAttention sparse range utilities for LocateAnything batch inference.""" |
| 2 | |
| 3 | from .range_attention import range_attention, is_available |
| 4 | |
| 5 | __all__ = ["range_attention", "is_available"] |
| 6 | |