[llvm] ae69f46 - [AbstractAttributor] Refine logic to indicate pessimistic fixed point when folding `__kmpc_is_spmd_exec_mode`

[Shilei Tian via llvm-commits]

Author: Shilei Tian
Date: 2021-07-23T13:36:47-04:00
New Revision: ae69f46867cd99d18ec7e23509eb1d71a098d48f

URL: https://github.com/llvm/llvm-project/commit/ae69f46867cd99d18ec7e23509eb1d71a098d48f
DIFF: https://github.com/llvm/llvm-project/commit/ae69f46867cd99d18ec7e23509eb1d71a098d48f.diff

LOG: [AbstractAttributor] Refine logic to indicate pessimistic fixed point when folding `__kmpc_is_spmd_exec_mode`

Since we are using assumed information now, the logic should be refined to avoid
unncessary assertion.

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D106630

Added: 
    

Modified: 
    llvm/lib/Transforms/IPO/OpenMPOpt.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
index 4532ab791eec..1031a55886d7 100644
--- a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
+++ b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
@@ -3698,10 +3698,8 @@ struct AAFoldRuntimeCallCallSiteReturned : AAFoldRuntimeCall {
       }
     }
 
-    if (KnownSPMDCount && KnownNonSPMDCount)
-      return indicatePessimisticFixpoint();
-
-    if (AssumedSPMDCount && AssumedNonSPMDCount)
+    if ((AssumedSPMDCount + KnownSPMDCount) &&
+        (AssumedNonSPMDCount + KnownNonSPMDCount))
       return indicatePessimisticFixpoint();
 
     auto &Ctx = getAnchorValue().getContext();