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()