From f2d9ad0620d9aa71481527dcfafdb8357da00470 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Mon, 10 Nov 2025 19:53:24 -0700 Subject: [PATCH] Only register rocm_aiter_ops if aiter is found (#28428) Signed-off-by: mgoin --- vllm/_aiter_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/_aiter_ops.py b/vllm/_aiter_ops.py index 9a4b5f3399bed..8d35aa65738b2 100644 --- a/vllm/_aiter_ops.py +++ b/vllm/_aiter_ops.py @@ -938,4 +938,5 @@ class rocm_aiter_ops: return tuple(shuffle_weight(tensor, layout=layout) for tensor in tensors) -rocm_aiter_ops.register_ops_once() +if IS_AITER_FOUND: + rocm_aiter_ops.register_ops_once()