Skip to content

Commit 6597d7a

Browse files
authored
[Platform] import activation_quant_fusion for CUDA only (vllm-project#23882)
Signed-off-by: wangxiyuan <[email protected]>
1 parent 5264015 commit 6597d7a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/compilation/pass_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,13 @@
88
from vllm.platforms import current_platform
99

1010
if current_platform.is_cuda_alike():
11+
from .activation_quant_fusion import ActivationQuantFusionPass
1112
from .fusion import FusionPass
1213
from .fusion_attn import AttnFusionPass
1314

1415
if current_platform.is_cuda():
1516
from .collective_fusion import AllReduceFusionPass, AsyncTPPass
1617

17-
from .activation_quant_fusion import ActivationQuantFusionPass
1818
from .fix_functionalization import FixFunctionalizationPass
1919
from .inductor_pass import CustomGraphPass, InductorPass, get_pass_context
2020
from .noop_elimination import NoOpEliminationPass

0 commit comments

Comments
 (0)