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: #906
rapidsai · Oct 7, 2022 · 942df21 · 942df21
1 parent a345a05
commit 942df21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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
+#endif