From 689962a060e443f1ffe17f3104b8ea2250422567 Mon Sep 17 00:00:00 2001 From: Dmitry Matveev Date: Wed, 2 Oct 2024 21:13:23 +0000 Subject: [PATCH] NPUW Spatial: Occasionally support spatial tail --- .../src/plugin/npuw/just_sync_infer_request.cpp | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/plugins/intel_npu/src/plugin/npuw/just_sync_infer_request.cpp b/src/plugins/intel_npu/src/plugin/npuw/just_sync_infer_request.cpp index 13e8e3f184c9aa..0a7f798c2aedef 100644 --- a/src/plugins/intel_npu/src/plugin/npuw/just_sync_infer_request.cpp +++ b/src/plugins/intel_npu/src/plugin/npuw/just_sync_infer_request.cpp @@ -440,11 +440,10 @@ void ov::npuw::JustInferRequest::bind_global_results(std::size_t idx) { LOG_BLOCK(); auto& comp_model_desc = m_npuw_model->m_compiled_submodels[idx]; - const auto real_idx = comp_model_desc.replaced_by.value_or(idx); - if (real_idx != idx) { - // Don't do here - function call will take parameter - // itself. Note it may be implemented more efficently - // than now (and in some cases, parameter can be pre-set) + if (comp_model_desc.replaced_by) { + // Don't do here - function call will take the right tensor + // itself. Note it may be implemented more efficently than now + // (and in some cases, the tensor can be pre-set) LOG_DEBUG("Skipping this too now - function will do it for itself"); return; }