From 462ea726866170894f1e617d034f511bb320d7bc Mon Sep 17 00:00:00 2001 From: Jaden Mathias Date: Thu, 19 Feb 2026 12:04:25 -0500 Subject: [PATCH] Add a aiter enabled check before appending unimported library --- vllm/compilation/passes/pass_manager.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/compilation/passes/pass_manager.py b/vllm/compilation/passes/pass_manager.py index 99bc37a0bbf4..21ecb41ded90 100644 --- a/vllm/compilation/passes/pass_manager.py +++ b/vllm/compilation/passes/pass_manager.py @@ -141,7 +141,8 @@ def configure(self, config: VllmConfig) -> None: if self.pass_config.fuse_rope_kvcache: self.passes += [SplitCoalescingPass(config)] self.passes += [ScatterSplitReplacementPass(config)] - self.passes += [ROCmAiterTritonRopeReshapeKVCacheFusionPass(config)] + if rocm_aiter_ops.is_enabled(): + self.passes += [ROCmAiterTritonRopeReshapeKVCacheFusionPass(config)] if self.pass_config.fuse_attn_quant: self.passes += [AttnFusionPass(config)]