[CPU] Add check for fused_input_proj in TP=4 (#35669)

2026-08-24 09:21:03 +08:00
parent e129fe21e5
commit 167c339c8e
1 changed files with 4 additions and 0 deletions
@@ -338,6 +338,10 @@ class Qwen3_5GatedDeltaNet(nn.Module):
                and self.in_proj_ba.bias is None
                and use_intel_amx_backend(self.in_proj_qkvz)
                and use_intel_amx_backend(self.in_proj_ba)
+                and (
+                    self.in_proj_qkvz.weight.size(0) % 32 == 0
+                    and self.in_proj_ba.weight.size(0) % 32 == 0
+                )
            )
        )