AMDGPU: Make sure both cc1 and cc1as process -m[no-]code-object-v3

[yaxunl]

clang/lib/Driver/ToolChains/Clang.cpp

6125

this does not work if you have multiple options, you may need something like

// Get the last argument of -mwavefrontsize64 or -mno-wavefrontsize64.
  if (auto *WaveArg = Args.getLastArg(options::OPT_mwavefrontsize64,
                                      options::OPT_mno_wavefrontsize64)) {
    if (WaveArg->getOption().getID() == options::OPT_mwavefrontsize64) {
    } else {
    }
  }

Also better extract this part as a function and call it in two places.

[scott.linder]

clang/test/Driver/amdgpu-features-as.s

1

Going along with Sam's request, can you add a test case for specifying multiple options?

[kzhuravl]

Addressed review feedback

[yaxunl]

LGTM

clang/lib/Driver/ToolChains/AMDGPU.cpp

void AMDGPUToolChain::addClangTargetOptions(
   // Default to "hidden" visibility, as object level linking will not be
   // supported for the foreseeable future.
   if (!DriverArgs.hasArg(options::OPT_fvisibility_EQ,
                          options::OPT_fvisibility_ms_compat)) {
     CC1Args.push_back("-fvisibility");
     CC1Args.push_back("hidden");
     CC1Args.push_back("-fapply-global-visibility-to-externs");
   }
-
-  if (DriverArgs.hasArg(options::OPT_mcode_object_v3_legacy)) {
-    getDriver().Diag(diag::warn_drv_deprecated_arg) << "-mcode-object-v3" <<
-      "-mllvm --amdhsa-code-object-version=3";
-    CC1Args.push_back("-mllvm");
-    CC1Args.push_back("--amdhsa-code-object-version=3");
-  }
-  if (DriverArgs.hasArg(options::OPT_mno_code_object_v3_legacy)) {
-    getDriver().Diag(diag::warn_drv_deprecated_arg) << "-mno-code-object-v3" <<
-      "-mllvm --amdhsa-code-object-version=2";
-    CC1Args.push_back("-mllvm");
-    CC1Args.push_back("--amdhsa-code-object-version=2");
-  }
 }
 
 StringRef
 AMDGPUToolChain::getGPUArch(const llvm::opt::ArgList &DriverArgs) const {
   return getProcessorFromTargetID(
       getTriple(), DriverArgs.getLastArgValue(options::OPT_mcpu_EQ));
 }
 

clang/lib/Driver/ToolChains/Clang.cpp

   if (IsOpenMPDevice) {
     CmdArgs.push_back("-fopenmp-is-device");
     if (OpenMPDeviceInput) {
       CmdArgs.push_back("-fopenmp-host-ir-file-path");
       CmdArgs.push_back(Args.MakeArgString(OpenMPDeviceInput->getFilename()));
     }
   }
 
+  if (Args.hasArg(options::OPT_mcode_object_v3_legacy)) {
+    D.Diag(diag::warn_drv_deprecated_arg) << "-mcode-object-v3" <<
+      "-mllvm --amdhsa-code-object-version=3";
+    CmdArgs.push_back("-mllvm");
+    CmdArgs.push_back("--amdhsa-code-object-version=3");
+  }
+  if (Args.hasArg(options::OPT_mno_code_object_v3_legacy)) {
+    D.Diag(diag::warn_drv_deprecated_arg) << "-mno-code-object-v3" <<
+      "-mllvm --amdhsa-code-object-version=2";
+    CmdArgs.push_back("-mllvm");
+    CmdArgs.push_back("--amdhsa-code-object-version=2");
+  }
+
   // For all the host OpenMP offloading compile jobs we need to pass the targets
   // information using -fopenmp-targets= option.
   if (JA.isHostOffloading(Action::OFK_OpenMP)) {
     SmallString<128> TargetInfo("-fopenmp-targets=");
 
     Arg *Tgts = Args.getLastArg(options::OPT_fopenmp_targets_EQ);
     assert(Tgts && Tgts->getNumValues() &&
            "OpenMP offloading has to have targets specified.");
void ClangAs::ConstructJob(Compilation &C, const JobAction &JA,
   const llvm::Triple &T = getToolChain().getTriple();
   Arg *A;
   if (getDebugFissionKind(D, Args, A) == DwarfFissionKind::Split &&
       T.isOSBinFormatELF()) {
     CmdArgs.push_back("-split-dwarf-output");
     CmdArgs.push_back(SplitDebugName(JA, Args, Input, Output));
   }
 
+  if (Args.hasArg(options::OPT_mcode_object_v3_legacy)) {
+    D.Diag(diag::warn_drv_deprecated_arg) << "-mcode-object-v3" <<
+      "-mllvm --amdhsa-code-object-version=3";
+    CmdArgs.push_back("-mllvm");
+    CmdArgs.push_back("--amdhsa-code-object-version=3");
+  }
+  if (Args.hasArg(options::OPT_mno_code_object_v3_legacy)) {
+    D.Diag(diag::warn_drv_deprecated_arg) << "-mno-code-object-v3" <<
+      "-mllvm --amdhsa-code-object-version=2";
+    CmdArgs.push_back("-mllvm");
+    CmdArgs.push_back("--amdhsa-code-object-version=2");
+  }
+
   assert(Input.isFilename() && "Invalid input.");
   CmdArgs.push_back(Input.getFilename());
 
   const char *Exec = getToolChain().getDriver().getClangProgramPath();
   C.addCommand(std::make_unique<Command>(JA, *this,
                                          ResponseFileSupport::AtFileUTF8(),
                                          Exec, CmdArgs, Inputs, Output));
 }

clang/test/Driver/amdgpu-features-as.s

+// RUN: %clang -### -target amdgcn-amd-amdhsa -mcpu=gfx900 -mcode-object-v3 %s 2>&1 | FileCheck --check-prefix=CODE-OBJECT-V3 %s
+// CODE-OBJECT-V3: warning: argument '-mcode-object-v3' is deprecated, use '-mllvm --amdhsa-code-object-version=3' instead [-Wdeprecated]
+// CODE-OBJECT-V3: "-mllvm" "--amdhsa-code-object-version=3"
+
+// RUN: %clang -### -target amdgcn-amd-amdhsa amdgcn -mcpu=gfx900 -mno-code-object-v3 %s 2>&1 | FileCheck --check-prefix=NO-CODE-OBJECT-V3 %s
+// NO-CODE-OBJECT-V3: warning: argument '-mno-code-object-v3' is deprecated, use '-mllvm --amdhsa-code-object-version=2' instead [-Wdeprecated]
+// NO-CODE-OBJECT-V3: "-mllvm" "--amdhsa-code-object-version=2"