File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change 55
66from vllm .config import VllmConfig
77from vllm .logger import init_logger
8+ from vllm .platforms import current_platform
9+
10+ if current_platform .is_cuda_alike ():
11+ from .fusion import FusionPass
12+ from .collective_fusion import AllReduceFusionPass , AsyncTPPass
13+ from .fusion_attn import AttnFusionPass
814
915from .activation_quant_fusion import ActivationQuantFusionPass
10- from .collective_fusion import AllReduceFusionPass , AsyncTPPass
1116from .fix_functionalization import FixFunctionalizationPass
12- from .fusion import FusionPass
13- from .fusion_attn import AttnFusionPass
1417from .inductor_pass import CustomGraphPass , InductorPass , get_pass_context
1518from .noop_elimination import NoOpEliminationPass
1619from .sequence_parallelism import SequenceParallelismPass
You can’t perform that action at this time.
0 commit comments