[PATCH] D108881: [clang][driver] Honor the last -flto= flag even if an earlier -flto is present

[Usman Nadeem via Phabricator via cfe-commits]

mnadeem created this revision.
mnadeem added a project: clang.
Herald added subscribers: ormris, steven_wu, hiraditya, inglorion.
mnadeem requested review of this revision.
Herald added a subscriber: cfe-commits.

After D102479 <https://reviews.llvm.org/D102479> `-flto=` options the end of the command line were being ignored if an earlier `-flto` was present.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D108881

Files:
  clang/lib/Driver/Driver.cpp
  clang/lib/Driver/ToolChains/Clang.cpp
  clang/test/Driver/lto.c


Index: clang/test/Driver/lto.c
===================================================================
--- clang/test/Driver/lto.c
+++ clang/test/Driver/lto.c
@@ -85,3 +85,15 @@
 // FLTO-AUTO: -flto=full
 // FLTO-JOBSERVER: -flto=full
 //
+
+// Pass the last -flto argument
+// RUN: %clang -target x86_64-unknown-linux -### %s -flto -flto=thin 2>&1 | \
+// RUN: FileCheck --check-prefix=FLTO-THIN %s
+// RUN: %clang -target x86_64-unknown-linux -### %s -flto=thin -flto=full 2>&1 | \
+// RUN: FileCheck --check-prefix=FLTO-FULL %s
+// RUN: %clang -target x86_64-unknown-linux -### %s -flto=thin -flto 2>&1 | \
+// RUN: FileCheck --check-prefix=FLTO %s
+//
+// FLTO-THIN: -flto=thin
+// FLTO-FULL: -flto=full
+// FLTO: "-flto"
Index: clang/lib/Driver/ToolChains/Clang.cpp
===================================================================
--- clang/lib/Driver/ToolChains/Clang.cpp
+++ clang/lib/Driver/ToolChains/Clang.cpp
@@ -4492,14 +4492,13 @@
 
     if (IsUsingLTO) {
       if (!IsDeviceOffloadAction) {
-        if (Args.hasArg(options::OPT_flto))
+        Arg *A = Args.getLastArg(options::OPT_flto, options::OPT_flto_EQ);
+        if (A && A->getOption().matches(options::OPT_flto))
           CmdArgs.push_back("-flto");
-        else {
-          if (D.getLTOMode() == LTOK_Thin)
-            CmdArgs.push_back("-flto=thin");
-          else
-            CmdArgs.push_back("-flto=full");
-        }
+        else if (D.getLTOMode() == LTOK_Thin)
+          CmdArgs.push_back("-flto=thin");
+        else
+          CmdArgs.push_back("-flto=full");
         CmdArgs.push_back("-flto-unit");
       } else if (Triple.isAMDGPU()) {
         // Only AMDGPU supports device-side LTO
Index: clang/lib/Driver/Driver.cpp
===================================================================
--- clang/lib/Driver/Driver.cpp
+++ clang/lib/Driver/Driver.cpp
@@ -600,8 +600,9 @@
 
   StringRef LTOName("full");
 
-  const Arg *A = Args.getLastArg(OptEq);
-  if (A)
+  const Arg *A = Args.getLastArg(OptEq, OptPos);
+  // Use the OptEq only if it is after the -flto option.
+  if (A && A->getOption().matches(OptEq))
     LTOName = A->getValue();
 
   LTOMode = llvm::StringSwitch<LTOKind>(LTOName)


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D108881.369289.patch
Type: text/x-patch
Size: 2190 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/cfe-commits/attachments/20210828/0975c266/attachment.bin>