1 //===--- AMDGPU.cpp - AMDGPU ToolChain Implementations ----------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "AMDGPU.h"
11 #include "CommonArgs.h"
12 #include "InputInfo.h"
13 #include "clang/Driver/Compilation.h"
14 #include "llvm/Option/ArgList.h"
15 
16 using namespace clang::driver;
17 using namespace clang::driver::tools;
18 using namespace clang::driver::toolchains;
19 using namespace clang;
20 using namespace llvm::opt;
21 
22 void amdgpu::Linker::ConstructJob(Compilation &C, const JobAction &JA,
23                                   const InputInfo &Output,
24                                   const InputInfoList &Inputs,
25                                   const ArgList &Args,
26                                   const char *LinkingOutput) const {
27 
28   std::string Linker = getToolChain().GetProgramPath(getShortName());
29   ArgStringList CmdArgs;
30   AddLinkerInputs(getToolChain(), Inputs, Args, CmdArgs, JA);
31   CmdArgs.push_back("-shared");
32   CmdArgs.push_back("-o");
33   CmdArgs.push_back(Output.getFilename());
34   C.addCommand(llvm::make_unique<Command>(JA, *this, Args.MakeArgString(Linker),
35                                           CmdArgs, Inputs));
36 }
37 
38 /// AMDGPU Toolchain
39 AMDGPUToolChain::AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple,
40                                  const ArgList &Args)
41     : Generic_ELF(D, Triple, Args),
42       OptionsDefault({{options::OPT_O, "3"},
43                       {options::OPT_cl_std_EQ, "CL1.2"}}) {}
44 
45 Tool *AMDGPUToolChain::buildLinker() const {
46   return new tools::amdgpu::Linker(*this);
47 }
48 
49 DerivedArgList *
50 AMDGPUToolChain::TranslateArgs(const DerivedArgList &Args, StringRef BoundArch,
51                                Action::OffloadKind DeviceOffloadKind) const {
52 
53   DerivedArgList *DAL =
54       Generic_ELF::TranslateArgs(Args, BoundArch, DeviceOffloadKind);
55 
56   // Do nothing if not OpenCL (-x cl)
57   if (!Args.getLastArgValue(options::OPT_x).equals("cl"))
58     return DAL;
59 
60   if (!DAL)
61     DAL = new DerivedArgList(Args.getBaseArgs());
62   for (auto *A : Args)
63     DAL->append(A);
64 
65   const OptTable &Opts = getDriver().getOpts();
66 
67   // Phase 1 (.cl -> .bc)
68   if (Args.hasArg(options::OPT_c) && Args.hasArg(options::OPT_emit_llvm)) {
69     DAL->AddFlagArg(nullptr, Opts.getOption(getTriple().isArch64Bit()
70                                                 ? options::OPT_m64
71                                                 : options::OPT_m32));
72 
73     // Have to check OPT_O4, OPT_O0 & OPT_Ofast separately
74     // as they defined that way in Options.td
75     if (!Args.hasArg(options::OPT_O, options::OPT_O0, options::OPT_O4,
76                      options::OPT_Ofast))
77       DAL->AddJoinedArg(nullptr, Opts.getOption(options::OPT_O),
78                         getOptionDefault(options::OPT_O));
79   }
80 
81   return DAL;
82 }
83