11
vllm_ascend/_310p/ops/fla/__init__.py
Normal file
11
vllm_ascend/_310p/ops/fla/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from .chunk_gated_delta_rule import chunk_gated_delta_rule_310
|
||||
from .fused_gdn_gating import fused_gdn_gating_pytorch
|
||||
from .fused_recurrent_gated_delta_rule import fused_recurrent_gated_delta_rule_pytorch
|
||||
from .l2norm import l2norm_310p
|
||||
|
||||
__all__ = [
|
||||
"fused_gdn_gating_pytorch",
|
||||
"fused_recurrent_gated_delta_rule_pytorch",
|
||||
"chunk_gated_delta_rule_310",
|
||||
"l2norm_310p",
|
||||
]
|
||||
Reference in New Issue
Block a user