File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
vllm/model_executor/layers/fused_moe Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change 12
12
import vllm .model_executor .layers .fused_moe .modular_kernel as mk
13
13
from vllm import _custom_ops as ops
14
14
from vllm .logger import init_logger
15
+ # yapf: disable
15
16
from vllm .model_executor .layers .fused_moe .config import (
16
17
FusedMoEQuantConfig , get_config_quant_dtype )
18
+ # yapf: enable
17
19
from vllm .model_executor .layers .fused_moe .deep_gemm_moe import (
18
20
_valid_deep_gemm , deep_gemm_moe_fp8 )
19
21
from vllm .model_executor .layers .fused_moe .moe_align_block_size import (
Original file line number Diff line number Diff line change 22
22
from vllm .forward_context import ForwardContext , get_forward_context
23
23
from vllm .logger import init_logger
24
24
from vllm .model_executor .custom_op import CustomOp
25
+ # yapf: disable
25
26
from vllm .model_executor .layers .fused_moe .config import (
26
27
FusedMoEConfig , FusedMoEParallelConfig )
28
+ # yapf: enable
27
29
from vllm .model_executor .layers .fused_moe .modular_kernel import (
28
30
FusedMoEActivationFormat , FusedMoEModularKernel ,
29
31
FusedMoEPermuteExpertsUnpermute , FusedMoEPrepareAndFinalize )
You can’t perform that action at this time.
0 commit comments