Commit 5c9e481d authored by shssf's avatar shssf Committed by Robert Kimball

IntelGPU backend: GetOutputElement operation fixed (#1624)

parent 5bcf3b5d
...@@ -73,8 +73,8 @@ static void arguments_check(const shared_ptr<Node>& op, size_t input, size_t out ...@@ -73,8 +73,8 @@ static void arguments_check(const shared_ptr<Node>& op, size_t input, size_t out
{ {
ostringstream os; ostringstream os;
os << "Operation \"" << op->description() << "\" input and output sizes mismatch." os << "Operation \"" << op->description() << "\" input and output sizes mismatch."
<< " Expected input size=" << op->get_input_size() << ", provided=" << input << " Expected input size=" << input << ", provided=" << op->get_input_size()
<< ". Expected output size=" << op->get_output_size() << ", provided=" << output; << ". Expected output size=" << output << ", provided=" << op->get_output_size();
throw invalid_argument(os.str()); throw invalid_argument(os.str());
} }
} }
...@@ -249,7 +249,10 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func) ...@@ -249,7 +249,10 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
} }
else if ("GetOutputElement" == op->description()) else if ("GetOutputElement" == op->description())
{ {
arguments_check(op, 3, 1); if (op->get_inputs().empty() || op->get_outputs().size() != 1)
{
arguments_check(op, 1, 1); // at least one input and exact one output expected
}
const shared_ptr<op::GetOutputElement> elem = const shared_ptr<op::GetOutputElement> elem =
static_pointer_cast<op::GetOutputElement>(op); static_pointer_cast<op::GetOutputElement>(op);
......
...@@ -12,6 +12,7 @@ backwards_dot_scalar_tensor ...@@ -12,6 +12,7 @@ backwards_dot_scalar_tensor
backwards_dot_tensor3_tensor3 backwards_dot_tensor3_tensor3
backwards_dot_tensor_scalar backwards_dot_tensor_scalar
backwards_dot_tensor_vector backwards_dot_tensor_vector
backwards_exp
backwards_floor backwards_floor
backwards_maxpool_n2_c1_hw5_3x3_str2_max backwards_maxpool_n2_c1_hw5_3x3_str2_max
backwards_maxpool_n4_c1_hw4_2x2_max backwards_maxpool_n4_c1_hw4_2x2_max
...@@ -21,11 +22,11 @@ backwards_reverse_sequence_n4d2c3h2w2 ...@@ -21,11 +22,11 @@ backwards_reverse_sequence_n4d2c3h2w2
backwards_sign backwards_sign
backwards_slice backwards_slice
backwards_tan backwards_tan
backwards_tanh
batch_norm_one_output batch_norm_one_output
batch_norm_three_outputs batch_norm_three_outputs
ceiling ceiling
concat_matrix_int64 concat_matrix_int64
convolution_outlining
divide_by_zero_int32 divide_by_zero_int32
dot_matrix_vector_int64 dot_matrix_vector_int64
floor floor
...@@ -52,6 +53,7 @@ replace_slice_3d ...@@ -52,6 +53,7 @@ replace_slice_3d
replace_slice_3d_strided replace_slice_3d_strided
replace_slice_3d_strided_different_strides replace_slice_3d_strided_different_strides
replace_slice_matrix replace_slice_matrix
replace_slice_matrix_inplace
replace_slice_scalar replace_slice_scalar
replace_slice_vector replace_slice_vector
reshape_6d reshape_6d
...@@ -63,6 +65,24 @@ select_and_scatter_with_overlap ...@@ -63,6 +65,24 @@ select_and_scatter_with_overlap
select_and_scatter_without_overlap select_and_scatter_without_overlap
sign sign
tan tan
topk_1d_max_all
topk_1d_max_one
topk_1d_max_partial
topk_1d_min_all
topk_1d_min_one
topk_1d_min_partial
topk_2d_max_all
topk_2d_max_one
topk_2d_max_partial
topk_2d_min_all
topk_2d_min_one
topk_2d_min_partial
topk_3d_max_all
topk_3d_max_one
topk_3d_max_partial
topk_3d_min_all
topk_3d_min_one
topk_3d_min_partial
zero_sized_abs zero_sized_abs
zero_sized_acos zero_sized_acos
zero_sized_add zero_sized_add
...@@ -92,23 +112,3 @@ zero_sized_sqrt ...@@ -92,23 +112,3 @@ zero_sized_sqrt
zero_sized_subtract zero_sized_subtract
zero_sized_tan zero_sized_tan
zero_sized_tanh zero_sized_tanh
argmin_trivial
argmax_trivial
topk_1d_max_all
topk_1d_max_partial
topk_1d_max_one
topk_1d_min_all
topk_1d_min_partial
topk_1d_min_one
topk_2d_max_all
topk_2d_max_partial
topk_2d_max_one
topk_2d_min_all
topk_2d_min_partial
topk_2d_min_one
topk_3d_max_all
topk_3d_max_partial
topk_3d_max_one
topk_3d_min_all
topk_3d_min_partial
topk_3d_min_one
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