Enabling CustomPasses for OpenCL

Conservatively clear the NSW NUW flags, since they may not be
preserved by the reassociation on CustomSafeOptPass

Author: agrabezh

IGC/AdaptorOCL/DriverInfoOCL.hpp

@@ -40,7 +40,7 @@ namespace TC
         bool enableVISAPreRAScheduler() const override { return true; }
 
         bool NeedWAToTransformA32MessagesToA64() const override{ return true; }
-        bool WADisableCustomPass() const override { return true; }
+        bool WADisableCustomPass() const override { return false; }
         bool WAEnableMemOpt2ForOCL() const override { return true; }
 
         unsigned int GetLoopUnrollThreshold() const override { return  1280; }

IGC/Compiler/CustomSafeOptPass.cpp

@@ -1379,6 +1379,14 @@ void CustomSafeOptPass::visitBinaryOperator(BinaryOperator& I)
                             if (nextInst->getOperand(i) == &I)
                             {
                                 Value* newAdd = BinaryOperator::CreateAdd(src0imm ? I.getOperand(1) : I.getOperand(0), nextInst->getOperand(1 - i), "", nextInst);
+
+                                // Conservatively clear the NSW NUW flags, since they may not be
+                                // preserved by the reassociation.
+                                bool IsNUW = isa<OverflowingBinaryOperator>(I) && I.hasNoUnsignedWrap() && nextInst->hasNoUnsignedWrap();
+                                cast<BinaryOperator>(newAdd)->setHasNoUnsignedWrap(IsNUW);
+                                nextInst->setHasNoUnsignedWrap(IsNUW);
+                                nextInst->setHasNoSignedWrap(false);
+
                                 nextInst->setOperand(0, newAdd);
                                 nextInst->setOperand(1, I.getOperand(src0imm ? 0 : 1));
                                 break;