InnerProductSpaceTraits::dot to compute the product

kokkos · Apr 25, 2018 · 1c68159 · 1c68159
1 parent 836726a
commit 1c68159
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/blas/impl/KokkosBlas1_team_dot_spec.hpp b/src/blas/impl/KokkosBlas1_team_dot_spec.hpp
@@ -62,16 +62,17 @@ struct team_dot_tpl_spec_avail {
 // Unification and Specialization layer
 template<class TeamType, class XV, class YV, bool tpl_spec_avail = team_dot_tpl_spec_avail<XV,YV>::value>
 struct TeamDot {
-  typedef typename Kokkos::Details::InnerProductSpaceTraits<typename XV::non_const_value_type>::dot_type dot_type;
   typedef          Kokkos::Details::InnerProductSpaceTraits<typename XV::non_const_value_type> IPT;
+  typedef typename IPT::dot_type dot_type;
 
   static KOKKOS_INLINE_FUNCTION dot_type team_dot (const TeamType& team, const XV& X, const YV& Y);
 };
 
 template<class TeamType, class XV, class YV>
 struct TeamDot<TeamType, XV, YV, false> {
-  typedef typename Kokkos::Details::InnerProductSpaceTraits<typename XV::non_const_value_type>::dot_type dot_type;
   typedef          Kokkos::Details::InnerProductSpaceTraits<typename XV::non_const_value_type> IPT;
+  typedef typename IPT::dot_type dot_type;
+
   static KOKKOS_INLINE_FUNCTION dot_type team_dot (const TeamType& team, const XV& X, const YV& Y) {
     dot_type result;
     int N = X.extent(0);