Update extensions.cpp

DeepLink-org · Mar 28, 2024 · c719183 · c719183
1 parent 78932e9
commit c719183
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/extensions.cpp b/csrc/extensions.cpp
@@ -71,7 +71,7 @@ auto extRmsNormBackward(const at::Tensor& input, const at::Tensor& grad_output,
                          std::move(grad_bias));
 }
 
-void extApplyRotary(at::Tensor output, const at::Tensor& input,
+void extApplyRotary(at::Tensor& output, const at::Tensor& input,
                     const at::Tensor& cos, const at::Tensor& sin,
                     const bool conj, const bool interleaved) {
   callDiopi(diopiRotaryEmbedding, output, input, cos, sin, conj, interleaved);