Fix handling saturation patterns

Author: krystian-andrzejewski

IGC/Compiler/CISACodeGen/PatternMatchPass.cpp

@@ -3219,6 +3219,19 @@ namespace IGC
             }
         };
 
+        // dp4a with modifiers
+        struct Dp4aSatPattern : Pattern
+        {
+            GenIntrinsicInst* inst;
+            virtual void Emit(EmitPass* pass, const DstModifier& modifier)
+            {
+                DstModifier mod = modifier;
+                mod.sat = true;
+                pass->emitDP4A(inst, nullptr, mod);
+            }
+        };
+
+
         bool match = false;
         llvm::Value* source = nullptr;
         bool isUnsigned = false;
@@ -3236,6 +3249,13 @@ namespace IGC
                     binaryOpInst && (binaryOpInst->getOpcode() == llvm::BinaryOperator::BinaryOps::Add) && isUnsigned)
                 {
                     match = true;
+
+                    uint numSources = GetNbSources(*sourceInst);
+                    for (uint i = 0; i < numSources; i++)
+                    {
+                        MarkAsSource(sourceInst->getOperand(i));
+                    }
+
                     UAddPattern* uAddPattern = new (m_allocator) UAddPattern();
                     uAddPattern->inst = binaryOpInst;
                     AddPattern(uAddPattern);
@@ -3264,9 +3284,16 @@ namespace IGC
                     genIsaInst->getIntrinsicID() == llvm::GenISAIntrinsic::ID::GenISA_dp4a_us))
                 {
                     match = true;
-                    SatPattern* satPattern = new (m_allocator) SatPattern();
-                    satPattern->pattern = Match(*sourceInst);
-                    AddPattern(satPattern);
+
+                    uint numSources = GetNbSources(*sourceInst);
+                    for (uint i = 0; i < numSources; i++)
+                    {
+                        MarkAsSource(sourceInst->getOperand(i));
+                    }
+
+                    Dp4aSatPattern* dp4aSatPattern = new (m_allocator) Dp4aSatPattern();
+                    dp4aSatPattern->inst = genIsaInst;
+                    AddPattern(dp4aSatPattern);
                 }
                 else
                 {

IGC/Compiler/CISACodeGen/opCode.h

@@ -116,7 +116,7 @@ DECLARE_OPCODE(SRem, Instruction, llvm_srem, true, false, false, false, false, f
 DECLARE_OPCODE(URem, Instruction, llvm_urem, true, false, false, false, false, false, false)
 DECLARE_OPCODE(SExt, Instruction, llvm_sext, true, true, false, false, false, false, false)
 DECLARE_OPCODE(ZExt, Instruction, llvm_zext, false, true, false, false, false, false, false)
-DECLARE_OPCODE(Trunc, Instruction, llvm_trunc, true, false, false, false, false, false, false)
+DECLARE_OPCODE(Trunc, Instruction, llvm_trunc, true, true, false, false, false, false, false)
 DECLARE_OPCODE(FPTrunc, Instruction, llvm_fptrunc, true, false, false, false, false, false, false)
 DECLARE_OPCODE(GenISA_ftof_rte, GenISAIntrinsic, llvm_fptrunc_rte, true, false, false, false, false, false, false)
 DECLARE_OPCODE(GenISA_ftof_rtz, GenISAIntrinsic, llvm_fptrunc_rtz, true, false, false, false, false, false, false)