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
{
ostringstream os;
os << "Operation \"" << op->description() << "\" input and output sizes mismatch."
<< " Expected input size=" << op->get_input_size() << ", provided=" << input
<< ". Expected output size=" << op->get_output_size() << ", provided=" << output;
<< " Expected input size=" << input << ", provided=" << op->get_input_size()
<< ". Expected output size=" << output << ", provided=" << op->get_output_size();
throw invalid_argument(os.str());
}
}
......@@ -249,7 +249,10 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
}
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 =
static_pointer_cast<op::GetOutputElement>(op);
......
......@@ -12,6 +12,7 @@ backwards_dot_scalar_tensor
backwards_dot_tensor3_tensor3
backwards_dot_tensor_scalar
backwards_dot_tensor_vector
backwards_exp
backwards_floor
backwards_maxpool_n2_c1_hw5_3x3_str2_max
backwards_maxpool_n4_c1_hw4_2x2_max
......@@ -21,11 +22,11 @@ backwards_reverse_sequence_n4d2c3h2w2
backwards_sign
backwards_slice
backwards_tan
backwards_tanh
batch_norm_one_output
batch_norm_three_outputs
ceiling
concat_matrix_int64
convolution_outlining
divide_by_zero_int32
dot_matrix_vector_int64
floor
......@@ -52,6 +53,7 @@ replace_slice_3d
replace_slice_3d_strided
replace_slice_3d_strided_different_strides
replace_slice_matrix
replace_slice_matrix_inplace
replace_slice_scalar
replace_slice_vector
reshape_6d
......@@ -63,6 +65,24 @@ select_and_scatter_with_overlap
select_and_scatter_without_overlap
sign
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_acos
zero_sized_add
......@@ -92,23 +112,3 @@ zero_sized_sqrt
zero_sized_subtract
zero_sized_tan
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