Updated introspection in feature extraction functor to correctly iden…

…tify feature vector on host
morpheus-org · Jun 7, 2023 · 839baea · 839baea
1 parent ec61590
commit 839baea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/MorpheusOracle_FeatureExtractionFunctors.hpp b/src/MorpheusOracle_FeatureExtractionFunctors.hpp
@@ -55,8 +55,8 @@ struct ExtractFeaturesFunctor {
       typename std::enable_if_t<
           Morpheus::is_dynamic_matrix_container_v<Matrix> &&
           Morpheus::is_dense_vector_format_container_v<FeatureVector> &&
-          Morpheus::has_access_v<ExecSpace, Matrix, FeatureVector>>* =
-          nullptr) {
+          Morpheus::has_host_memory_space_v<FeatureVector> &&
+          Morpheus::has_access_v<ExecSpace, Matrix>>* = nullptr) {
     static_assert(
         std::is_floating_point_v<typename FeatureVector::value_type>,
         "The value_type of the FeatureVector must be a floating-point "