Fix DCOFF closure being allocated in the bank

smirnov-alexey · Oct 9, 2024 · 322cc2f · 322cc2f
1 parent 348730e
commit 322cc2f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/plugins/intel_npu/src/plugin/npuw/partitioning/patterns/dcoff.cpp b/src/plugins/intel_npu/src/plugin/npuw/partitioning/patterns/dcoff.cpp
@@ -125,7 +125,8 @@ void apply_remap(Subgraph& fcall, const ClosureRemap& m) {
     // empty tensors by default.
     for (auto&& i : m.closure_remap) {
         new_lazy_closure.push_back(fcall._lazy_closure[i]);
-        new_closure.push_back(fcall._closure[i]);
+        // _closure is empty at this stage. To fill it for DCOFF, evaluate _lazy_closure
+        new_closure.push_back(fcall._lazy_closure[i].eval());
 
         auto scale_iter = m.scale_remap.find(i);
         auto zerop_iter = m.zerop_remap.find(i);