Files
xc-llm-ascend/vllm_ascend/_310p/ops/fla/__init__.py

10 lines
342 B
Python
Raw Permalink Normal View History

from .chunk_gated_delta_rule import chunk_gated_delta_rule_pytorch
from .fused_gdn_gating import fused_gdn_gating_pytorch
from .fused_recurrent_gated_delta_rule import fused_recurrent_gated_delta_rule_pytorch
__all__ = [
"fused_gdn_gating_pytorch",
"fused_recurrent_gated_delta_rule_pytorch",
"chunk_gated_delta_rule_pytorch",
]