Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / api / CPP / program.hpp

diff --git a/inference-engine/thirdparty/clDNN/api/CPP/program.hpp b/inference-engine/thirdparty/clDNN/api/CPP/program.hpp
index 6657765..a8520ad 100644 (file)
--- a/inference-engine/thirdparty/clDNN/api/CPP/program.hpp
+++ b/inference-engine/thirdparty/clDNN/api/CPP/program.hpp
@@ -41,6 +41,9 @@ enum class build_option_type
     /// @brief Enable implicit reordering for user inputs (default: false).
     optimize_data = cldnn_build_option_optimize_data,
 
+    /// @brief Enable running detection output layer always on gpu, regardless performance
+    detection_output_gpu = cldnn_build_option_detection_output_gpu,
+
     /// @brief Enable debug mode (default: false).
     /// @details This option enforce all program primitives to be accessible as outputs.
     debug = cldnn_build_option_debug,
@@ -112,6 +115,9 @@ struct build_option
     /// @brief Enable implicit reordering for user inputs (default: false).
     static std::shared_ptr<const build_option> optimize_data(bool enable = false);
 
+    /// @brief Enable running detection output layer always on GPU, regardless performance (default: false).
+    static std::shared_ptr<const build_option> detection_output_gpu(bool enable = false);
+
     /// @brief Enable debug mode (default: false).
     /// @details This option enforce all program primitives to be accessible as outputs.
     static std::shared_ptr<const build_option> debug(bool enable = false);
@@ -462,6 +468,16 @@ namespace detail
             return std::make_shared<object_type>(option);
         }
     };
+    template<> struct build_option_traits<build_option_type::detection_output_gpu>
+    {
+        typedef build_option_bool<build_option_type::detection_output_gpu> object_type;
+        static std::shared_ptr<const build_option> make_default() { return build_option::detection_output_gpu(); }
+        static std::shared_ptr<const build_option> make_option(const cldnn_build_option& option)
+        {
+            assert(option.type == cldnn_build_option_detection_output_gpu);
+            return std::make_shared<object_type>(option);
+        }
+    };
     template<> struct build_option_traits<build_option_type::debug>
     {
         typedef build_option_bool<build_option_type::debug> object_type;
@@ -547,6 +563,11 @@ inline std::shared_ptr<const build_option> build_option::optimize_data(bool enab
     return std::make_shared<build_option_bool<build_option_type::optimize_data>>(enable);
 }
 
+inline std::shared_ptr<const build_option> build_option::detection_output_gpu(bool enable)
+{
+    return std::make_shared<build_option_bool<build_option_type::detection_output_gpu>>(enable);
+}
+
 inline std::shared_ptr<const build_option> build_option::debug(bool enable)
 {
     return std::make_shared<build_option_bool<build_option_type::debug>>(enable);
@@ -664,10 +685,12 @@ private:
         {
         case cldnn_build_option_fusing:
             return detail::build_option_traits<build_option_type::fusing>::make_option(option);
-               case cldnn_build_option_learning_config:
-                       return detail::build_option_traits<build_option_type::learning_config>::make_option(option);
+        case cldnn_build_option_learning_config:
+            return detail::build_option_traits<build_option_type::learning_config>::make_option(option);
         case cldnn_build_option_optimize_data:
             return detail::build_option_traits<build_option_type::optimize_data>::make_option(option);
+        case cldnn_build_option_detection_output_gpu:
+            return detail::build_option_traits<build_option_type::detection_output_gpu>::make_option(option);
         case cldnn_build_option_debug:
             return detail::build_option_traits<build_option_type::debug>::make_option(option);
         case cldnn_build_option_outputs: