You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
364 B
18 lines
364 B
4 days ago
|
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()
|