From ee965c9c69f7cf287a1ec530390a9a9fc3748273 Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Sun, 22 Dec 2024 22:16:12 -0800
Subject: [PATCH] Use default

Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
---
 vllm/_custom_ops.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py
index b5b0115f4cf3b..415345cdc7b6b 100644
--- a/vllm/_custom_ops.py
+++ b/vllm/_custom_ops.py
@@ -256,8 +256,8 @@ def copy_subranges(
     tgt_matrix: torch.Tensor,
     num_subranges: int,
 ) -> None:
-    torch.ops._C.copy_subranges(src_matrix, diff_matrix, tgt_matrix,
-                                num_subranges)
+    torch.ops._C.copy_subranges.default(src_matrix, diff_matrix, tgt_matrix,
+                                        num_subranges)
 
 
 # fused quant layer norm ops