| 1234567891011121314151617 |
- import functools
- from torch.utils._triton import has_triton
- @functools.cache
- def has_helion_package() -> bool:
- try:
- import helion # type: ignore[import-untyped, import-not-found] # noqa: F401
- except ImportError:
- return False
- return True
- @functools.cache
- def has_helion() -> bool:
- return has_helion_package() and has_triton()
|
