提交vllm0.11.0开发分支
This commit is contained in:
9
vllm_kunlun/ops/fla/__init__.py
Normal file
9
vllm_kunlun/ops/fla/__init__.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from .chunk import chunk_gated_delta_rule
|
||||
from .fused_recurrent import fused_recurrent_gated_delta_rule
|
||||
from .layernorm_guard import RMSNormGated
|
||||
from .torch_fla import l2norm, torch_chunk_gated_delta_rule
|
||||
__all__ = [
|
||||
"RMSNormGated",
|
||||
"chunk_gated_delta_rule",
|
||||
"fused_recurrent_gated_delta_rule",
|
||||
]
|
||||
Reference in New Issue
Block a user