avoiding unnecessary copying

hellopahe · Skylion007 · web-flow · commit 691c5d5e3c11 · 2025-02-07T01:30:09.000+08:00
Co-authored-by: Aaron Gokaslan &lt;aaronGokaslan@gmail.com&gt;
diff --git a/aten/src/ATen/native/mps/operations/Attention.mm b/aten/src/ATen/native/mps/operations/Attention.mm
@@ -27,7 +27,7 @@
         t = x.unsqueeze(0);
         squeezed = true;
     }
-    return {t, squeezed};
+    return {std::move(t), squeezed};
 }
 
 std::tuple<Tensor, Tensor> _scaled_dot_product_attention_math_mps(const Tensor& query,

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@`
`27`	`27`	`t = x.unsqueeze(0);`
`28`	`28`	`squeezed = true;`
`29`	`29`	`}`
`30`		`- return {t, squeezed};`
	`30`	`+ return {std::move(t), squeezed};`
`31`	`31`	`}`
`32`	`32`
`33`	`33`	`std::tuple<Tensor, Tensor> _scaled_dot_product_attention_math_mps(const Tensor& query,`