eckit::linalg::dense::LinearAlgebraTorch completion

pmaciel · pmaciel · commit 1123fb3af63d · 2026-02-23T10:29:11.000Z
diff --git a/src/eckit/linalg/dense/LinearAlgebraTorch.cc b/src/eckit/linalg/dense/LinearAlgebraTorch.cc
@@ -38,7 +38,7 @@ Scalar LinearAlgebraTorch::dot(const Vector& x, const Vector& y) const {
     auto x_tensor = make_dense_tensor(x);
     auto y_tensor = make_dense_tensor(y);
 
-    return torch::dot(x_tensor, y_tensor).to(torch::DeviceType::CPU, torch::kFloat64).item<Scalar>();
+    return tensor_to_host(torch::dot(x_tensor, y_tensor)).item<Scalar>();
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ Scalar LinearAlgebraTorch::dot(const Vector& x, const Vector& y) const {`
`38`	`38`	`auto x_tensor = make_dense_tensor(x);`
`39`	`39`	`auto y_tensor = make_dense_tensor(y);`
`40`	`40`
`41`		`- return torch::dot(x_tensor, y_tensor).to(torch::DeviceType::CPU, torch::kFloat64).item<Scalar>();`
	`41`	`+ return tensor_to_host(torch::dot(x_tensor, y_tensor)).item<Scalar>();`
`42`	`42`	`}`
`43`	`43`
`44`	`44`