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, ©_subranges); // Layernorm