natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Fix transposed shape calculation for static model (#24329) 

### Details:
- Fix transposed shape calculation for static model where the transposed
order is less then canonicalized shape

### Tickets:
 - 140091
This commit is contained in:
Taylor Yeonbok Lee 2024-05-01 02:47:19 -07:00 committed by GitHub
parent 179fde64d8
commit ac27cd6fe7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/plugins/intel_gpu/src/graph/impls/ocl/gemm.cpp
View File

@ -188,11 +188,20 @@ public:
(primitive->input_rank == primitive->weight_rank);
if (is_broadcastable) {
auto transpose_pshape = [](const ov::PartialShape pshape, const std::vector<int64_t>& order) {
auto transposed_pshape = ov::PartialShape::dynamic(pshape.rank());
for (size_t i = 0; i < order.size(); i++) {
transposed_pshape[i] = pshape[order[i]];
if (order.size() < pshape.size()) {
auto transposed_pshape = pshape;
auto rank_diff = pshape.size() - order.size();
for (size_t i = 0; i < order.size(); i++) {
transposed_pshape[i + rank_diff] = pshape[rank_diff + order[i]];
}
return transposed_pshape;
} else {
auto transposed_pshape = ov::PartialShape::dynamic(pshape.rank());
for (size_t i = 0; i < order.size(); i++) {
transposed_pshape[i] = pshape[order[i]];
}
return transposed_pshape;
}
return transposed_pshape;
};
size_t max_rank = input0_pshape.size();
auto default_order = ov::intel_gpu::op::Gemm::default_order(max_rank);