File tree Expand file tree Collapse file tree 3 files changed +13
-14
lines changed
tests/unit/transformations Expand file tree Collapse file tree 3 files changed +13
-14
lines changed Original file line number Diff line number Diff line change @@ -688,8 +688,7 @@ void FullyConnected::needUpdateTensorParalelConfig() {
688688 tp_cfg.enable_tensor_parallel = false ;
689689 return ;
690690 }
691- const auto rank = shape.getRank ();
692- if (rank == 3 && shape.getDims ()[2 ] > 1 ) {
691+ if (shape.getRank () == 3 && shape.getDims ().back () > 1 ) {
693692 tp_cfg.enable_tensor_parallel = false ;
694693 return ;
695694 }
Original file line number Diff line number Diff line change 55#include < gtest/gtest.h>
66
77#include < memory>
8- #include < openvino/core/model.hpp>
8+ #include " openvino/core/model.hpp"
99#include " openvino/opsets/opset1_decl.hpp"
10- #include < openvino/pass/manager.hpp>
11- #include < ov_ops/type_relaxed.hpp>
12- #include < transformations/cpu_opset/common/pass/convert_matmul_to_fc.hpp>
13- #include < transformations/init_node_info.hpp>
14- #include < transformations/utils/utils.hpp>
10+ #include " openvino/pass/manager.hpp"
11+ #include " ov_ops/type_relaxed.hpp"
12+ #include " transformations/cpu_opset/common/pass/convert_matmul_to_fc.hpp"
13+ #include " transformations/init_node_info.hpp"
14+ #include " transformations/utils/utils.hpp"
1515
1616#include " common_test_utils/ov_test_utils.hpp"
1717#include " openvino/op/constant.hpp"
Original file line number Diff line number Diff line change 55#include < gtest/gtest.h>
66
77#include < memory>
8- #include < openvino/core/model.hpp>
8+ #include " openvino/core/model.hpp"
99#include " openvino/opsets/opset1_decl.hpp"
10- #include < openvino/pass/manager.hpp>
11- #include < ov_ops/type_relaxed.hpp>
12- #include < transformations/cpu_opset/common/pass/convert_matmul_to_fc.hpp>
13- #include < transformations/init_node_info.hpp>
14- #include < transformations/utils/utils.hpp>
10+ #include " openvino/pass/manager.hpp"
11+ #include " ov_ops/type_relaxed.hpp"
12+ #include " transformations/cpu_opset/common/pass/convert_matmul_to_fc.hpp"
13+ #include " transformations/init_node_info.hpp"
14+ #include " transformations/utils/utils.hpp"
1515
1616#include " common_test_utils/ov_test_utils.hpp"
1717#include " openvino/op/constant.hpp"
You can’t perform that action at this time.
0 commit comments