We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5dd4f33 commit 9630fa7Copy full SHA for 9630fa7
src/plugins/intel_gpu/src/graph/moe_mask_gen.cpp
@@ -90,8 +90,7 @@ std::vector<layout> moe_mask_gen_reshape_inst::calc_output_layouts(moe_mask_gen_
90
// tokens_per_expert
91
output_layouts.emplace_back(impl_param.get_input_layout(0));
92
// 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);
+ output_layouts.emplace_back(impl_param.get_input_layout(1));
95
// experts_id
96
auto experts_ids_shape = ov::Shape{static_cast<size_t>(num_actually_used_experts)};
97
output_layouts.emplace_back(experts_ids_shape, data_types::i32, format::bfyx);
0 commit comments