Skip to content

Commit

Permalink
[GPU] Fixed unit tests affected by skipping reorder opt
Browse files Browse the repository at this point in the history
  • Loading branch information
steve-y committed Nov 21, 2024
1 parent e3001e0 commit 1d15768
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -192,9 +192,9 @@ TEST(add_required_reorders, skip_adding_reorder_batch_axis_padding) {
crop_prim = network.get_primitive("crop2");
ASSERT_EQ(crop_prim->can_be_optimized(), true);
auto reorder_prim = network.get_primitive("crop1_reorder");
ASSERT_EQ(reorder_prim->can_be_optimized(), true);
ASSERT_EQ(reorder_prim->can_be_optimized(), false);
reorder_prim = network.get_primitive("crop2_reorder");
ASSERT_EQ(reorder_prim->can_be_optimized(), true);
ASSERT_EQ(reorder_prim->can_be_optimized(), false);
auto concate = network.get_primitive("concat");
ASSERT_EQ(concate->can_be_optimized(), false);
}
Original file line number Diff line number Diff line change
Expand Up @@ -1174,7 +1174,7 @@ TEST(prepare_buffer_fusing, test_implicit_crop_and_outerpadding) {
auto reorder_prim = network.get_primitive("gather1_reorder");
ASSERT_EQ(reorder_prim->can_be_optimized(), true);
reorder_prim = network.get_primitive("gather2_reorder");
ASSERT_EQ(reorder_prim->can_be_optimized(), true);
ASSERT_EQ(reorder_prim->can_be_optimized(), false);
auto reshape_prim = network.get_primitive("reshape1");
ASSERT_EQ(reshape_prim->can_be_optimized(), true);
}
Expand Down

0 comments on commit 1d15768

Please sign in to comment.