Skip to content

Conversation

@yuanxion
Copy link
Contributor

@yuanxion yuanxion commented Dec 3, 2025

Details

  • Changed to get single dim from partial_shape when get_dims is not available for dynamic layout

Description of the issue

Symptom

E2E test for Pytorch_YoloV9_api_2_True will be failed at "get_dims()" for "__module.model.22/aten::cat/Concat_4" node during prepare_buffer_fusing pass.

Root cause

"__module.model.22/aten::cat/Concat_4" node has 2 dependent nodes as its inputs. The layout of 1 of the dependent node is dynamic and used to call "get_dims()", which caused the issue.

How to fix it

The concat_axis dim in dims is not dynamic, and can be used to get_length().

The code and line that caused this issue

  • get_dims

auto input_length = pred_layout.get_dims()[concat_axis];

Reproduction step and snapshot

python -m pytest test_ovc_mo.py --tb=native --env_conf=.automation/env_config.yml --test_conf=test_config_local.yml -m "not launch_only_if_manually_specified " --pregen_irs=python_api_serialized/irs_mapping.csv --tf_models_version=1.15.2 --modules pipelines/production/pytorch/heavy/yolov9.py -k "Pytorch_YoloV9_api_2_True" --dynamism_type=None

Problematic graph

  • get_dims with dynamic layout (dep 1)
image

Checklist

  • Is it a proper fix? (not a workaround)
  • Did you include test case for this fix, if necessary?
  • Did you review existing test that can be extended to cover this scenario? Which test did you review?

Tickets:

@yuanxion yuanxion requested review from a team as code owners December 3, 2025 09:19
@github-actions github-actions bot added the category: GPU OpenVINO GPU plugin label Dec 3, 2025
@yuanxion yuanxion marked this pull request as draft December 4, 2025 05:18
@yuanxion yuanxion added this to the 2025.4 milestone Dec 4, 2025
@yuanxion yuanxion force-pushed the fix-dynamic-shape-get-dims branch from e665608 to 53ee06a Compare December 4, 2025 13:09
@yuanxion yuanxion marked this pull request as ready for review December 4, 2025 13:10
@yuanxion yuanxion marked this pull request as draft December 5, 2025 05:08
Signed-off-by: yuan.xiong <yuan.xiong@intel.com>
Signed-off-by: yuan.xiong <yuan.xiong@intel.com>
Signed-off-by: yuan.xiong <yuan.xiong@intel.com>
Signed-off-by: yuan.xiong <yuan.xiong@intel.com>
@yuanxion yuanxion force-pushed the fix-dynamic-shape-get-dims branch from 71e7f88 to 117ad05 Compare December 7, 2025 14:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

category: GPU OpenVINO GPU plugin

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants