Skip to content

Commit 5e2ee32

Browse files
Add allow_new_shape_infer false condition instead of checking whether static shape
1 parent fc17499 commit 5e2ee32

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_buffer_fusing.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -852,7 +852,8 @@ void prepare_buffer_fusing::run(program& p) {
852852
// Therefore, the padding of crop_layout should be shifted properly.
853853
const size_t TDIM = 4;
854854
auto user = node.get_users().front();
855-
if (user->is_type<gemm>() && pred_layout.is_static() && user->get_dependency(1).id().compare(node.id()) == 0) {
855+
bool allow_new_shape_infer = node.get_program().is_new_shape_infer();
856+
if (!allow_new_shape_infer && user->is_type<gemm>() && user->get_dependency(1).id().compare(node.id()) == 0) {
856857
auto input_rank = user->get_kernel_impl_params()->typed_desc<gemm>()->weight_rank;
857858
if (input_rank < TDIM) {
858859
std::vector<int32_t> l_pad = {0, 0, 0, 0};

0 commit comments

Comments
 (0)