From 0a669eed7ba600f26b98c45710af29db707ba4b5 Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Sat, 21 Dec 2024 17:39:13 -0800
Subject: [PATCH] Minor

Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
---
 csrc/torch_bindings.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/csrc/torch_bindings.cpp b/csrc/torch_bindings.cpp
index f418b580d8225..31394c04597a5 100644
--- a/csrc/torch_bindings.cpp
+++ b/csrc/torch_bindings.cpp
@@ -104,8 +104,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
 
   ops.def(
       "copy_subranges(Tensor matrix_src, Tensor matrix_diff, Tensor! "
-      "matrix_tgt, "
-      "int n) -> ()");
+      "matrix_tgt, int n) -> ()");
   ops.impl("copy_subranges", torch::kCUDA, &copy_subranges);
 
   // Layernorm