[NPUW] Fix scales processing in CWAI for nf4 (#28523)

### Tickets: - *E-149709*
openvinotoolkit · Jan 17, 2025 · 509ce8b · 509ce8b
1 parent a506af0
commit 509ce8b
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
@@ -890,7 +890,8 @@ CWAI3::CWAI3(CWAI3::Results scales) {
         auto matched_valueA = std::static_pointer_cast<ov::op::v0::Constant>(matched_nodeA);
         auto matched_valueC = std::static_pointer_cast<ov::op::v0::Constant>(matched_nodeC);
 
-        if (ov::element::i4 == matched_valueA->get_element_type() &&
+        if ((ov::element::i4 == matched_valueA->get_element_type() ||
+             ov::element::nf4 == matched_valueA->get_element_type()) &&
             (ov::element::f16 == matched_valueC->get_element_type() ||
              ov::element::f32 == matched_valueC->get_element_type())) {
             LOG_DEBUG("Matched: " << matched_valueC);