From 942df21d632d847ba7aae7e25b1198e8a41f25af Mon Sep 17 00:00:00 2001
From: "Corey J. Nolet" <cjnolet@users.noreply.github.com>
Date: Fri, 7 Oct 2022 10:34:06 -0400
Subject: [PATCH] Adding missing include for device mdspan in
 `mean_squared_error.cuh` (#906)

Authors:
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Divye Gala (https://github.com/divyegala)

URL: https://github.com/rapidsai/raft/pull/906
---
 cpp/include/raft/linalg/detail/mean_squared_error.cuh | 2 +-
 cpp/include/raft/linalg/mean_squared_error.cuh        | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/cpp/include/raft/linalg/detail/mean_squared_error.cuh b/cpp/include/raft/linalg/detail/mean_squared_error.cuh
index 5889314eea..9cb2c0354b 100644
--- a/cpp/include/raft/linalg/detail/mean_squared_error.cuh
+++ b/cpp/include/raft/linalg/detail/mean_squared_error.cuh
@@ -30,7 +30,7 @@ void meanSquaredError(
     math_t diff = a - b;
     return diff * diff * weight / len;
   };
-  raft::linalg::mapThenSumReduce<math_t, decltype(sq_diff), TPB>(out, len, sq_diff, stream, A, B);
+  raft::linalg::mapThenSumReduce<math_t, decltype(sq_diff)>(out, len, sq_diff, stream, A, B);
 }
 
 };  // end namespace detail
diff --git a/cpp/include/raft/linalg/mean_squared_error.cuh b/cpp/include/raft/linalg/mean_squared_error.cuh
index 582bab2acc..298f32339c 100644
--- a/cpp/include/raft/linalg/mean_squared_error.cuh
+++ b/cpp/include/raft/linalg/mean_squared_error.cuh
@@ -19,6 +19,7 @@
 #pragma once
 
 #include "detail/mean_squared_error.cuh"
+#include <raft/core/device_mdspan.hpp>
 
 namespace raft {
 namespace linalg {
@@ -76,4 +77,4 @@ void mean_squared_error(const raft::handle_t& handle,
 };  // end namespace linalg
 };  // end namespace raft
 
-#endif
\ No newline at end of file
+#endif