Skip to content

Commit 83a4e78

Browse files
committed
[gpu]:[istft]: Fixed issue with constant input shape.
1 parent e9b30fe commit 83a4e78

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/plugins/intel_gpu/src/plugin/program_builder.cpp

-3
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,6 @@
1010
#include "openvino/op/lstm_cell.hpp"
1111
#include "openvino/op/lstm_sequence.hpp"
1212
#include "openvino/op/loop.hpp"
13-
#include "openvino/op/search_sorted.hpp"
14-
#include "openvino/op/stft.hpp"
15-
#include "openvino/op/istft.hpp"
1613
#include "ov_ops/dynamic_quantize.hpp"
1714
#include "openvino/runtime/properties.hpp"
1815

src/plugins/intel_gpu/src/runtime/execution_config.cpp

+3-2
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
#include "openvino/op/paged_attention.hpp"
1212
#include "openvino/op/search_sorted.hpp"
1313
#include "openvino/op/stft.hpp"
14+
#include "openvino/op/istft.hpp"
1415
#include "ov_ops/dynamic_quantize.hpp"
1516
#include "ov_ops/rms.hpp"
1617
#include "openvino/runtime/internal_properties.hpp"
@@ -44,8 +45,8 @@ bool requires_new_shape_infer(const std::shared_ptr<ov::Node>& op) {
4445
// HACK: SearchSorted has specific shape requirements.
4546
// E.g. static input shapes: sorted:[8], values:[2,3,4] are prefectly fine,
4647
// but sorted:[8,1,1,1], values:[2,3,4,1] is not valid.
47-
// Similar case for STFT.
48-
if (ov::is_type<ov::op::v15::SearchSorted>(op) || ov::is_type<ov::op::v15::STFT>(op))
48+
// Similar case for STFT and ISTFT
49+
if (ov::is_type<ov::op::v15::SearchSorted>(op) || ov::is_type<ov::op::v15::STFT>(op) || ov::is_type<ov::op::v16::ISTFT>(op))
4950
return true;
5051

5152
if (ov::is_type<ov::op::internal::DynamicQuantize>(op) || ov::is_type<ov::op::internal::RMS>(op))

0 commit comments

Comments
 (0)