Only register rocm_aiter_ops if aiter is found (#28428)

Signed-off-by: mgoin <mgoin64@gmail.com>
2025-11-10 19:53:24 -07:00
parent de540c0354
commit f2d9ad0620
1 changed files with 2 additions and 1 deletions
--- 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()