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 @@
#include <CPP/reorder.hpp>
#include <CPP/reshape.hpp>
#include <CPP/scale.hpp>
#include <CPP/select.hpp>
#include <CPP/softmax.hpp>
#include <CPP/topology.hpp>
......@@ -75,6 +76,8 @@ using namespace ngraph;
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:
// Abs,
// Acos,
......@@ -346,6 +349,8 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
{
arguments_check(op, 3, 1);
// Leave it here for some time
#if USE_INTELGPU_CUSTOM_KERNELS
do_select_operation(topology,
get_input_name(op, 0),
get_input_shape(op, 0),
......@@ -356,6 +361,13 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
get_output_name(op),
get_output_shape(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;
}
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