Skip to content

Commit 9630fa7

Browse files
committed
updated moe_mask_gen_reshape not to update experts_info_start_idx
1 parent 5dd4f33 commit 9630fa7

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/plugins/intel_gpu/src/graph/moe_mask_gen.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,7 @@ std::vector<layout> moe_mask_gen_reshape_inst::calc_output_layouts(moe_mask_gen_
9090
// tokens_per_expert
9191
output_layouts.emplace_back(impl_param.get_input_layout(0));
9292
// experts_info_start_idx
93-
auto experts_info_start_idx_shape = ov::Shape{static_cast<size_t>(num_actually_used_experts)};
94-
output_layouts.emplace_back(experts_info_start_idx_shape, data_types::i32, format::bfyx);
93+
output_layouts.emplace_back(impl_param.get_input_layout(1));
9594
// experts_id
9695
auto experts_ids_shape = ov::Shape{static_cast<size_t>(num_actually_used_experts)};
9796
output_layouts.emplace_back(experts_ids_shape, data_types::i32, format::bfyx);

0 commit comments

Comments
 (0)