Unverified Commit 5cdddddd authored by Yanan Cao's avatar Yanan Cao Committed by GitHub
Browse files

[Kernel] [Helion] Force disable HOP path due to performance regression (#40171)


Signed-off-by: default avatarYanan Cao <gmagogsfm@gmail.com>
Co-authored-by: default avatarClaude Sonnet 4 <noreply@anthropic.com>
parent 6ef1efd5
...@@ -53,14 +53,15 @@ if not has_helion(): ...@@ -53,14 +53,15 @@ if not has_helion():
) )
import helion import helion
from helion._compat import requires_torch_version
from helion.autotuner.base_search import BaseAutotuner from helion.autotuner.base_search import BaseAutotuner
from helion.runtime.config import Config from helion.runtime.config import Config
from helion.runtime.settings import default_autotuner_fn from helion.runtime.settings import default_autotuner_fn
# TODO(gmagogsfm): Remove CustomOp fallback path (_get_or_register_custom_op, # TODO(gmagogsfm): Remove CustomOp fallback path (_get_or_register_custom_op,
# vllm_helion_lib, direct_register_custom_op) once vLLM requires PyTorch >= 2.11. # vllm_helion_lib, direct_register_custom_op) once vLLM requires PyTorch >= 2.11.
_HOP_AVAILABLE = requires_torch_version("2.11") # FIXME(gmagogsfm): Re-enable HOP path once performance regression is fixed.
# _HOP_AVAILABLE = requires_torch_version("2.11")
_HOP_AVAILABLE = False
if _HOP_AVAILABLE: if _HOP_AVAILABLE:
from helion._compat import supports_torch_compile_fusion from helion._compat import supports_torch_compile_fusion
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment