IntelGPU backend: Change exception type for unsupported op (#1957)

0ad9ec2c · Sergey Shalnov · Robert Kimball · 5b994011 · 0ad9ec2c
Commit 0ad9ec2c authored 6 years ago by Sergey Shalnov Committed by Robert Kimball 6 years ago
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

intelgpu_backend.cpp src/ngraph/runtime/intelgpu/intelgpu_backend.cpp +1 -4

No files found.
--- a/src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
+++ b/src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
@@ -432,10 +432,6 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
            function_output_names.push_back(get_input_name(op));
            break;
        }
-        case OP_TYPEID::GenerateMask:
-        {
-            throw ngraph_error("GenerateMask isn't yet supported on integrated GPU");
-        }
        case OP_TYPEID::GetOutputElement:
        {
            if (op->get_inputs().empty() || op->get_outputs().size() != 1)
@@ -1508,6 +1504,7 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
        case OP_TYPEID::Quantize:
        case OP_TYPEID::ReduceWindow:
        case OP_TYPEID::ReplaceSlice:
+        case OP_TYPEID::GenerateMask:
        case OP_TYPEID::ReverseSequence:
        case OP_TYPEID::SelectAndScatter:
        case OP_TYPEID::StopGradient: