Commit 3db4f121 authored by dmyershov's avatar dmyershov Committed by Robert Kimball

IntelGPU backend: Switch Select operation to use implementation from … (#1688)

* IntelGPU backend: Switch Select operation to use implementation from clDNN

* PR1688: Style fix.
parent de9a9346
...@@ -29,6 +29,7 @@ ...@@ -29,6 +29,7 @@
#include <CPP/reorder.hpp> #include <CPP/reorder.hpp>
#include <CPP/reshape.hpp> #include <CPP/reshape.hpp>
#include <CPP/scale.hpp> #include <CPP/scale.hpp>
#include <CPP/select.hpp>
#include <CPP/softmax.hpp> #include <CPP/softmax.hpp>
#include <CPP/topology.hpp> #include <CPP/topology.hpp>
...@@ -75,6 +76,8 @@ using namespace ngraph; ...@@ -75,6 +76,8 @@ using namespace ngraph;
using intelgpu_space = runtime::intelgpu::IntelGPULayout; using intelgpu_space = runtime::intelgpu::IntelGPULayout;
#define USE_INTELGPU_CUSTOM_KERNELS 0
// This expands the op list in op_tbl.hpp into a list of enumerations that look like this: // This expands the op list in op_tbl.hpp into a list of enumerations that look like this:
// Abs, // Abs,
// Acos, // Acos,
...@@ -346,6 +349,8 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func) ...@@ -346,6 +349,8 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
{ {
arguments_check(op, 3, 1); arguments_check(op, 3, 1);
// Leave it here for some time
#if USE_INTELGPU_CUSTOM_KERNELS
do_select_operation(topology, do_select_operation(topology,
get_input_name(op, 0), get_input_name(op, 0),
get_input_shape(op, 0), get_input_shape(op, 0),
...@@ -356,6 +361,13 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func) ...@@ -356,6 +361,13 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
get_output_name(op), get_output_name(op),
get_output_shape(op), get_output_shape(op),
get_output_type(op)); get_output_type(op));
#else
const cldnn::select cldnn_select(get_output_name(op),
get_input_name(op, 1),
get_input_name(op, 2),
get_input_name(op));
topology.add(cldnn_select);
#endif
break; break;
} }
case OP_TYPEID::Reverse: case OP_TYPEID::Reverse:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment