Skip to content

Commit

Permalink
[GPU] Fix random tests
Browse files Browse the repository at this point in the history
  • Loading branch information
p-durandin committed Sep 9, 2024
1 parent c5892bd commit c4f2fdb
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ TEST(prepare_buffer_fusing, optimize_reshape) {

auto input_memory = engine.allocate_memory(layout{ ov::PartialShape{1, 2, 2, 4}, data_types::f16, format::bfyx });
auto pattern_memory = engine.allocate_memory(layout{ ov::PartialShape{4}, data_types::i64, format::bfyx });
set_values<float>(input_memory, {0.1, 1.1, 2.2, 3.0, 4.0, -5.0, 0.1, 0.7, 4.8, 19.2, -10.1, 8.1, 10.2, 1.3, 1.44, 1.5});
set_values<ov::float16>(input_memory, {0.1, 1.1, 2.2, 3.0, 4.0, -5.0, 0.1, 0.7, 4.8, 19.2, -10.1, 8.1, 10.2, 1.3, 1.44, 1.5});
set_values<int64_t>(pattern_memory, {1, 4, 1, -1});

net.set_input_data("input", input_memory);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -370,7 +370,7 @@ TEST(prepare_primitive_fusing, fuse_eltwise_to_fc_dyn_illegal_2) {

auto input_memory = engine.allocate_memory(layout{ ov::PartialShape{1, 10}, data_types::i8, format::bfyx });
auto extra_input_memory = engine.allocate_memory(layout{ ov::PartialShape{4, 4}, data_types::f32, format::bfyx });
set_values<int8_t>(input_memory, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -10});
set_values<int8_t>(input_memory, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9});
set_values<float>(extra_input_memory, {1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4});

net.set_input_data("input", input_memory);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1884,7 +1884,7 @@ TEST(border_gpu, basic_zero_input) {
layout zero_input_layout = {{0, 1}, data_types::f32, format::bfyx};
input = engine.reinterpret_buffer(*input, zero_input_layout);

ov::CoordinateDiff pads_begin = {4, 0};
std::vector<int> pads_begin = {4, 0};
ov::PartialShape pads_begin_shape = { ov::Dimension(pads_begin.size()) };
auto pads_begin_input = engine.allocate_memory({pads_begin_shape, data_types::i32, format::bfyx});
set_values(pads_begin_input, pads_begin);
Expand Down

0 comments on commit c4f2fdb

Please sign in to comment.