[SpeculativeExecution] Add only-if-divergent-target pass option

The optimization pipeline enables this option, but it was not
preserved in -print-pipeline-passes output.

Author: nikic

llvm/include/llvm/Transforms/Scalar/SpeculativeExecution.h

@@ -73,6 +73,9 @@ class SpeculativeExecutionPass
 
   PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM);
 
+  void printPipeline(raw_ostream &OS,
+                     function_ref<StringRef(StringRef)> MapClassName2PassName);
+
   // Glue for old PM
   bool runImpl(Function &F, TargetTransformInfo *TTI);
 

llvm/lib/Passes/PassBuilder.cpp

@@ -1082,6 +1082,11 @@ Expected<bool> parseMemorySSAPrinterPassOptions(StringRef Params) {
                                "MemorySSAPrinterPass");
 }
 
+Expected<bool> parseSpeculativeExecutionPassOptions(StringRef Params) {
+  return parseSinglePassOption(Params, "only-if-divergent-target",
+                               "SpeculativeExecutionPass");
+}
+
 Expected<std::string> parseMemProfUsePassOptions(StringRef Params) {
   std::string Result;
   while (!Params.empty()) {

llvm/lib/Passes/PassRegistry.def

@@ -424,7 +424,6 @@ FUNCTION_PASS("sccp", SCCPPass())
 FUNCTION_PASS("sink", SinkingPass())
 FUNCTION_PASS("slp-vectorizer", SLPVectorizerPass())
 FUNCTION_PASS("slsr", StraightLineStrengthReducePass())
-FUNCTION_PASS("speculative-execution", SpeculativeExecutionPass())
 FUNCTION_PASS("strip-gc-relocates", StripGCRelocates())
 FUNCTION_PASS("structurizecfg", StructurizeCFGPass())
 FUNCTION_PASS("tailcallelim", TailCallElimPass())
@@ -588,6 +587,13 @@ FUNCTION_PASS_WITH_PARAMS("print<memoryssa>",
                            },
                           parseMemorySSAPrinterPassOptions,
                           "no-ensure-optimized-uses")
+FUNCTION_PASS_WITH_PARAMS("speculative-execution",
+                          "SpeculativeExecutionPass",
+                          [](bool OnlyIfDivergentTarget) {
+                            return SpeculativeExecutionPass(OnlyIfDivergentTarget);
+                          },
+                          parseSpeculativeExecutionPassOptions,
+                          "only-if-divergent-target")
 #undef FUNCTION_PASS_WITH_PARAMS
 
 #ifndef LOOPNEST_PASS

llvm/lib/Transforms/Scalar/SpeculativeExecution.cpp

@@ -346,4 +346,14 @@ PreservedAnalyses SpeculativeExecutionPass::run(Function &F,
   PA.preserveSet<CFGAnalyses>();
   return PA;
 }
+
+void SpeculativeExecutionPass::printPipeline(
+    raw_ostream &OS, function_ref<StringRef(StringRef)> MapClassName2PassName) {
+  static_cast<PassInfoMixin<SpeculativeExecutionPass> *>(this)->printPipeline(
+      OS, MapClassName2PassName);
+  OS << '<';
+  if (OnlyIfDivergentTarget)
+    OS << "only-if-divergent-target";
+  OS << '>';
+}
 }  // namespace llvm

llvm/test/CodeGen/NVPTX/speculative-execution-divergent-target.ll

@@ -6,8 +6,13 @@
 ; RUN: opt < %s -S -mtriple=nvptx-nvidia-cuda -passes=speculative-execution \
 ; RUN:   -spec-exec-only-if-divergent-target | \
 ; RUN:   FileCheck --check-prefix=ON %s
+; RUN: opt < %s -S -mtriple=nvptx-nvidia-cuda \
+; RUN:   -passes='speculative-execution<only-if-divergent-target>' | \
+; RUN:   FileCheck --check-prefix=ON %s
 ; RUN: opt < %s -S -passes=speculative-execution -spec-exec-only-if-divergent-target | \
 ; RUN:   FileCheck --check-prefix=OFF %s
+; RUN: opt < %s -S -passes='speculative-execution<only-if-divergent-target>' | \
+; RUN:   FileCheck --check-prefix=OFF %s
 
 ; Hoist in if-then pattern.
 define void @f() {

llvm/test/Other/new-pm-print-pipeline.ll

@@ -120,3 +120,6 @@
 
 ; RUN: opt -disable-output -disable-verify -print-pipeline-passes -passes='globaldce<vfe-linkage-unit-visibility>' < %s | FileCheck %s --match-full-lines --check-prefixes=CHECK-35
 ; CHECK-35: globaldce<vfe-linkage-unit-visibility>
+
+; RUN: opt -disable-output -disable-verify -print-pipeline-passes -passes='speculative-execution<only-if-divergent-target>' < %s | FileCheck %s --match-full-lines --check-prefixes=CHECK-36
+; CHECK-36: function(speculative-execution<only-if-divergent-target>)