1 //===--- Cuda.cpp - Cuda Tool and 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 "Cuda.h" 11 #include "InputInfo.h" 12 #include "CommonArgs.h" 13 #include "clang/Basic/Cuda.h" 14 #include "clang/Config/config.h" 15 #include "clang/Basic/VirtualFileSystem.h" 16 #include "clang/Driver/Compilation.h" 17 #include "clang/Driver/Driver.h" 18 #include "clang/Driver/DriverDiagnostic.h" 19 #include "clang/Driver/Options.h" 20 #include "llvm/Option/ArgList.h" 21 #include "llvm/Support/Path.h" 22 #include <system_error> 23 24 using namespace clang::driver; 25 using namespace clang::driver::toolchains; 26 using namespace clang::driver::tools; 27 using namespace clang; 28 using namespace llvm::opt; 29 30 // Parses the contents of version.txt in an CUDA installation. It should 31 // contain one line of the from e.g. "CUDA Version 7.5.2". 32 static CudaVersion ParseCudaVersionFile(llvm::StringRef V) { 33 if (!V.startswith("CUDA Version ")) 34 return CudaVersion::UNKNOWN; 35 V = V.substr(strlen("CUDA Version ")); 36 int Major = -1, Minor = -1; 37 auto First = V.split('.'); 38 auto Second = First.second.split('.'); 39 if (First.first.getAsInteger(10, Major) || 40 Second.first.getAsInteger(10, Minor)) 41 return CudaVersion::UNKNOWN; 42 43 if (Major == 7 && Minor == 0) { 44 // This doesn't appear to ever happen -- version.txt doesn't exist in the 45 // CUDA 7 installs I've seen. But no harm in checking. 46 return CudaVersion::CUDA_70; 47 } 48 if (Major == 7 && Minor == 5) 49 return CudaVersion::CUDA_75; 50 if (Major == 8 && Minor == 0) 51 return CudaVersion::CUDA_80; 52 return CudaVersion::UNKNOWN; 53 } 54 55 CudaInstallationDetector::CudaInstallationDetector( 56 const Driver &D, const llvm::Triple &HostTriple, 57 const llvm::opt::ArgList &Args) 58 : D(D) { 59 SmallVector<std::string, 4> CudaPathCandidates; 60 61 // In decreasing order so we prefer newer versions to older versions. 62 std::initializer_list<const char *> Versions = {"8.0", "7.5", "7.0"}; 63 64 if (Args.hasArg(clang::driver::options::OPT_cuda_path_EQ)) { 65 CudaPathCandidates.push_back( 66 Args.getLastArgValue(clang::driver::options::OPT_cuda_path_EQ)); 67 } else if (HostTriple.isOSWindows()) { 68 for (const char *Ver : Versions) 69 CudaPathCandidates.push_back( 70 D.SysRoot + "/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v" + 71 Ver); 72 } else { 73 CudaPathCandidates.push_back(D.SysRoot + "/usr/local/cuda"); 74 for (const char *Ver : Versions) 75 CudaPathCandidates.push_back(D.SysRoot + "/usr/local/cuda-" + Ver); 76 } 77 78 for (const auto &CudaPath : CudaPathCandidates) { 79 if (CudaPath.empty() || !D.getVFS().exists(CudaPath)) 80 continue; 81 82 InstallPath = CudaPath; 83 BinPath = CudaPath + "/bin"; 84 IncludePath = InstallPath + "/include"; 85 LibDevicePath = InstallPath + "/nvvm/libdevice"; 86 87 auto &FS = D.getVFS(); 88 if (!(FS.exists(IncludePath) && FS.exists(BinPath) && 89 FS.exists(LibDevicePath))) 90 continue; 91 92 // On Linux, we have both lib and lib64 directories, and we need to choose 93 // based on our triple. On MacOS, we have only a lib directory. 94 // 95 // It's sufficient for our purposes to be flexible: If both lib and lib64 96 // exist, we choose whichever one matches our triple. Otherwise, if only 97 // lib exists, we use it. 98 if (HostTriple.isArch64Bit() && FS.exists(InstallPath + "/lib64")) 99 LibPath = InstallPath + "/lib64"; 100 else if (FS.exists(InstallPath + "/lib")) 101 LibPath = InstallPath + "/lib"; 102 else 103 continue; 104 105 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> VersionFile = 106 FS.getBufferForFile(InstallPath + "/version.txt"); 107 if (!VersionFile) { 108 // CUDA 7.0 doesn't have a version.txt, so guess that's our version if 109 // version.txt isn't present. 110 Version = CudaVersion::CUDA_70; 111 } else { 112 Version = ParseCudaVersionFile((*VersionFile)->getBuffer()); 113 } 114 115 std::error_code EC; 116 for (llvm::sys::fs::directory_iterator LI(LibDevicePath, EC), LE; 117 !EC && LI != LE; LI = LI.increment(EC)) { 118 StringRef FilePath = LI->path(); 119 StringRef FileName = llvm::sys::path::filename(FilePath); 120 // Process all bitcode filenames that look like libdevice.compute_XX.YY.bc 121 const StringRef LibDeviceName = "libdevice."; 122 if (!(FileName.startswith(LibDeviceName) && FileName.endswith(".bc"))) 123 continue; 124 StringRef GpuArch = FileName.slice( 125 LibDeviceName.size(), FileName.find('.', LibDeviceName.size())); 126 LibDeviceMap[GpuArch] = FilePath.str(); 127 // Insert map entries for specifc devices with this compute 128 // capability. NVCC's choice of the libdevice library version is 129 // rather peculiar and depends on the CUDA version. 130 if (GpuArch == "compute_20") { 131 LibDeviceMap["sm_20"] = FilePath; 132 LibDeviceMap["sm_21"] = FilePath; 133 LibDeviceMap["sm_32"] = FilePath; 134 } else if (GpuArch == "compute_30") { 135 LibDeviceMap["sm_30"] = FilePath; 136 if (Version < CudaVersion::CUDA_80) { 137 LibDeviceMap["sm_50"] = FilePath; 138 LibDeviceMap["sm_52"] = FilePath; 139 LibDeviceMap["sm_53"] = FilePath; 140 } 141 LibDeviceMap["sm_60"] = FilePath; 142 LibDeviceMap["sm_61"] = FilePath; 143 LibDeviceMap["sm_62"] = FilePath; 144 } else if (GpuArch == "compute_35") { 145 LibDeviceMap["sm_35"] = FilePath; 146 LibDeviceMap["sm_37"] = FilePath; 147 } else if (GpuArch == "compute_50") { 148 if (Version >= CudaVersion::CUDA_80) { 149 LibDeviceMap["sm_50"] = FilePath; 150 LibDeviceMap["sm_52"] = FilePath; 151 LibDeviceMap["sm_53"] = FilePath; 152 } 153 } 154 } 155 156 // This code prevents IsValid from being set when 157 // no libdevice has been found. 158 bool allEmpty = true; 159 std::string LibDeviceFile; 160 for (auto key : LibDeviceMap.keys()) { 161 LibDeviceFile = LibDeviceMap.lookup(key); 162 if (!LibDeviceFile.empty()) 163 allEmpty = false; 164 } 165 166 if (allEmpty) 167 continue; 168 169 IsValid = true; 170 break; 171 } 172 } 173 174 void CudaInstallationDetector::AddCudaIncludeArgs( 175 const ArgList &DriverArgs, ArgStringList &CC1Args) const { 176 if (!DriverArgs.hasArg(options::OPT_nobuiltininc)) { 177 // Add cuda_wrappers/* to our system include path. This lets us wrap 178 // standard library headers. 179 SmallString<128> P(D.ResourceDir); 180 llvm::sys::path::append(P, "include"); 181 llvm::sys::path::append(P, "cuda_wrappers"); 182 CC1Args.push_back("-internal-isystem"); 183 CC1Args.push_back(DriverArgs.MakeArgString(P)); 184 } 185 186 if (DriverArgs.hasArg(options::OPT_nocudainc)) 187 return; 188 189 if (!isValid()) { 190 D.Diag(diag::err_drv_no_cuda_installation); 191 return; 192 } 193 194 CC1Args.push_back("-internal-isystem"); 195 CC1Args.push_back(DriverArgs.MakeArgString(getIncludePath())); 196 CC1Args.push_back("-include"); 197 CC1Args.push_back("__clang_cuda_runtime_wrapper.h"); 198 } 199 200 void CudaInstallationDetector::CheckCudaVersionSupportsArch( 201 CudaArch Arch) const { 202 if (Arch == CudaArch::UNKNOWN || Version == CudaVersion::UNKNOWN || 203 ArchsWithVersionTooLowErrors.count(Arch) > 0) 204 return; 205 206 auto RequiredVersion = MinVersionForCudaArch(Arch); 207 if (Version < RequiredVersion) { 208 ArchsWithVersionTooLowErrors.insert(Arch); 209 D.Diag(diag::err_drv_cuda_version_too_low) 210 << InstallPath << CudaArchToString(Arch) << CudaVersionToString(Version) 211 << CudaVersionToString(RequiredVersion); 212 } 213 } 214 215 void CudaInstallationDetector::print(raw_ostream &OS) const { 216 if (isValid()) 217 OS << "Found CUDA installation: " << InstallPath << ", version " 218 << CudaVersionToString(Version) << "\n"; 219 } 220 221 void NVPTX::Assembler::ConstructJob(Compilation &C, const JobAction &JA, 222 const InputInfo &Output, 223 const InputInfoList &Inputs, 224 const ArgList &Args, 225 const char *LinkingOutput) const { 226 const auto &TC = 227 static_cast<const toolchains::CudaToolChain &>(getToolChain()); 228 assert(TC.getTriple().isNVPTX() && "Wrong platform"); 229 230 StringRef GPUArchName; 231 // If this is an OpenMP action we need to extract the device architecture 232 // from the -march=arch option. This option may come from -Xopenmp-target 233 // flag or the default value. 234 if (JA.isDeviceOffloading(Action::OFK_OpenMP)) { 235 GPUArchName = Args.getLastArgValue(options::OPT_march_EQ); 236 assert(!GPUArchName.empty() && "Must have an architecture passed in."); 237 } else 238 GPUArchName = JA.getOffloadingArch(); 239 240 // Obtain architecture from the action. 241 CudaArch gpu_arch = StringToCudaArch(GPUArchName); 242 assert(gpu_arch != CudaArch::UNKNOWN && 243 "Device action expected to have an architecture."); 244 245 // Check that our installation's ptxas supports gpu_arch. 246 if (!Args.hasArg(options::OPT_no_cuda_version_check)) { 247 TC.CudaInstallation.CheckCudaVersionSupportsArch(gpu_arch); 248 } 249 250 ArgStringList CmdArgs; 251 CmdArgs.push_back(TC.getTriple().isArch64Bit() ? "-m64" : "-m32"); 252 if (Args.hasFlag(options::OPT_cuda_noopt_device_debug, 253 options::OPT_no_cuda_noopt_device_debug, false)) { 254 // ptxas does not accept -g option if optimization is enabled, so 255 // we ignore the compiler's -O* options if we want debug info. 256 CmdArgs.push_back("-g"); 257 CmdArgs.push_back("--dont-merge-basicblocks"); 258 CmdArgs.push_back("--return-at-end"); 259 } else if (Arg *A = Args.getLastArg(options::OPT_O_Group)) { 260 // Map the -O we received to -O{0,1,2,3}. 261 // 262 // TODO: Perhaps we should map host -O2 to ptxas -O3. -O3 is ptxas's 263 // default, so it may correspond more closely to the spirit of clang -O2. 264 265 // -O3 seems like the least-bad option when -Osomething is specified to 266 // clang but it isn't handled below. 267 StringRef OOpt = "3"; 268 if (A->getOption().matches(options::OPT_O4) || 269 A->getOption().matches(options::OPT_Ofast)) 270 OOpt = "3"; 271 else if (A->getOption().matches(options::OPT_O0)) 272 OOpt = "0"; 273 else if (A->getOption().matches(options::OPT_O)) { 274 // -Os, -Oz, and -O(anything else) map to -O2, for lack of better options. 275 OOpt = llvm::StringSwitch<const char *>(A->getValue()) 276 .Case("1", "1") 277 .Case("2", "2") 278 .Case("3", "3") 279 .Case("s", "2") 280 .Case("z", "2") 281 .Default("2"); 282 } 283 CmdArgs.push_back(Args.MakeArgString(llvm::Twine("-O") + OOpt)); 284 } else { 285 // If no -O was passed, pass -O0 to ptxas -- no opt flag should correspond 286 // to no optimizations, but ptxas's default is -O3. 287 CmdArgs.push_back("-O0"); 288 } 289 290 // Pass -v to ptxas if it was passed to the driver. 291 if (Args.hasArg(options::OPT_v)) 292 CmdArgs.push_back("-v"); 293 294 CmdArgs.push_back("--gpu-name"); 295 CmdArgs.push_back(Args.MakeArgString(CudaArchToString(gpu_arch))); 296 CmdArgs.push_back("--output-file"); 297 SmallString<256> OutputFileName(Output.getFilename()); 298 if (JA.isOffloading(Action::OFK_OpenMP)) 299 llvm::sys::path::replace_extension(OutputFileName, "cubin"); 300 CmdArgs.push_back(Args.MakeArgString(OutputFileName)); 301 for (const auto& II : Inputs) 302 CmdArgs.push_back(Args.MakeArgString(II.getFilename())); 303 304 for (const auto& A : Args.getAllArgValues(options::OPT_Xcuda_ptxas)) 305 CmdArgs.push_back(Args.MakeArgString(A)); 306 307 // In OpenMP we need to generate relocatable code. 308 if (JA.isOffloading(Action::OFK_OpenMP) && 309 Args.hasFlag(options::OPT_fopenmp_relocatable_target, 310 options::OPT_fnoopenmp_relocatable_target, 311 /*Default=*/ true)) 312 CmdArgs.push_back("-c"); 313 314 const char *Exec; 315 if (Arg *A = Args.getLastArg(options::OPT_ptxas_path_EQ)) 316 Exec = A->getValue(); 317 else 318 Exec = Args.MakeArgString(TC.GetProgramPath("ptxas")); 319 C.addCommand(llvm::make_unique<Command>(JA, *this, Exec, CmdArgs, Inputs)); 320 } 321 322 // All inputs to this linker must be from CudaDeviceActions, as we need to look 323 // at the Inputs' Actions in order to figure out which GPU architecture they 324 // correspond to. 325 void NVPTX::Linker::ConstructJob(Compilation &C, const JobAction &JA, 326 const InputInfo &Output, 327 const InputInfoList &Inputs, 328 const ArgList &Args, 329 const char *LinkingOutput) const { 330 const auto &TC = 331 static_cast<const toolchains::CudaToolChain &>(getToolChain()); 332 assert(TC.getTriple().isNVPTX() && "Wrong platform"); 333 334 ArgStringList CmdArgs; 335 CmdArgs.push_back("--cuda"); 336 CmdArgs.push_back(TC.getTriple().isArch64Bit() ? "-64" : "-32"); 337 CmdArgs.push_back(Args.MakeArgString("--create")); 338 CmdArgs.push_back(Args.MakeArgString(Output.getFilename())); 339 340 for (const auto& II : Inputs) { 341 auto *A = II.getAction(); 342 assert(A->getInputs().size() == 1 && 343 "Device offload action is expected to have a single input"); 344 const char *gpu_arch_str = A->getOffloadingArch(); 345 assert(gpu_arch_str && 346 "Device action expected to have associated a GPU architecture!"); 347 CudaArch gpu_arch = StringToCudaArch(gpu_arch_str); 348 349 // We need to pass an Arch of the form "sm_XX" for cubin files and 350 // "compute_XX" for ptx. 351 const char *Arch = 352 (II.getType() == types::TY_PP_Asm) 353 ? CudaVirtualArchToString(VirtualArchForCudaArch(gpu_arch)) 354 : gpu_arch_str; 355 CmdArgs.push_back(Args.MakeArgString(llvm::Twine("--image=profile=") + 356 Arch + ",file=" + II.getFilename())); 357 } 358 359 for (const auto& A : Args.getAllArgValues(options::OPT_Xcuda_fatbinary)) 360 CmdArgs.push_back(Args.MakeArgString(A)); 361 362 const char *Exec = Args.MakeArgString(TC.GetProgramPath("fatbinary")); 363 C.addCommand(llvm::make_unique<Command>(JA, *this, Exec, CmdArgs, Inputs)); 364 } 365 366 void NVPTX::OpenMPLinker::ConstructJob(Compilation &C, const JobAction &JA, 367 const InputInfo &Output, 368 const InputInfoList &Inputs, 369 const ArgList &Args, 370 const char *LinkingOutput) const { 371 const auto &TC = 372 static_cast<const toolchains::CudaToolChain &>(getToolChain()); 373 assert(TC.getTriple().isNVPTX() && "Wrong platform"); 374 375 ArgStringList CmdArgs; 376 377 // OpenMP uses nvlink to link cubin files. The result will be embedded in the 378 // host binary by the host linker. 379 assert(!JA.isHostOffloading(Action::OFK_OpenMP) && 380 "CUDA toolchain not expected for an OpenMP host device."); 381 382 if (Output.isFilename()) { 383 CmdArgs.push_back("-o"); 384 CmdArgs.push_back(Output.getFilename()); 385 } else 386 assert(Output.isNothing() && "Invalid output."); 387 if (Args.hasArg(options::OPT_g_Flag)) 388 CmdArgs.push_back("-g"); 389 390 if (Args.hasArg(options::OPT_v)) 391 CmdArgs.push_back("-v"); 392 393 StringRef GPUArch = 394 Args.getLastArgValue(options::OPT_march_EQ); 395 assert(!GPUArch.empty() && "At least one GPU Arch required for ptxas."); 396 397 CmdArgs.push_back("-arch"); 398 CmdArgs.push_back(Args.MakeArgString(GPUArch)); 399 400 // Add paths specified in LIBRARY_PATH environment variable as -L options. 401 addDirectoryList(Args, CmdArgs, "-L", "LIBRARY_PATH"); 402 403 // Add paths for the default clang library path. 404 SmallString<256> DefaultLibPath = 405 llvm::sys::path::parent_path(TC.getDriver().Dir); 406 llvm::sys::path::append(DefaultLibPath, "lib" CLANG_LIBDIR_SUFFIX); 407 CmdArgs.push_back(Args.MakeArgString(Twine("-L") + DefaultLibPath)); 408 409 // Add linking against library implementing OpenMP calls on NVPTX target. 410 CmdArgs.push_back("-lomptarget-nvptx"); 411 412 for (const auto &II : Inputs) { 413 if (II.getType() == types::TY_LLVM_IR || 414 II.getType() == types::TY_LTO_IR || 415 II.getType() == types::TY_LTO_BC || 416 II.getType() == types::TY_LLVM_BC) { 417 C.getDriver().Diag(diag::err_drv_no_linker_llvm_support) 418 << getToolChain().getTripleString(); 419 continue; 420 } 421 422 // Currently, we only pass the input files to the linker, we do not pass 423 // any libraries that may be valid only for the host. 424 if (!II.isFilename()) 425 continue; 426 427 SmallString<256> Name = llvm::sys::path::filename(II.getFilename()); 428 llvm::sys::path::replace_extension(Name, "cubin"); 429 430 const char *CubinF = 431 C.addTempFile(C.getArgs().MakeArgString(Name)); 432 433 CmdArgs.push_back(CubinF); 434 } 435 436 AddOpenMPLinkerScript(getToolChain(), C, Output, Inputs, Args, CmdArgs, JA); 437 438 const char *Exec = 439 Args.MakeArgString(getToolChain().GetProgramPath("nvlink")); 440 C.addCommand(llvm::make_unique<Command>(JA, *this, Exec, CmdArgs, Inputs)); 441 } 442 443 /// CUDA toolchain. Our assembler is ptxas, and our "linker" is fatbinary, 444 /// which isn't properly a linker but nonetheless performs the step of stitching 445 /// together object files from the assembler into a single blob. 446 447 CudaToolChain::CudaToolChain(const Driver &D, const llvm::Triple &Triple, 448 const ToolChain &HostTC, const ArgList &Args, 449 const Action::OffloadKind OK) 450 : ToolChain(D, Triple, Args), HostTC(HostTC), 451 CudaInstallation(D, HostTC.getTriple(), Args), OK(OK) { 452 if (CudaInstallation.isValid()) 453 getProgramPaths().push_back(CudaInstallation.getBinPath()); 454 // Lookup binaries into the driver directory, this is used to 455 // discover the clang-offload-bundler executable. 456 getProgramPaths().push_back(getDriver().Dir); 457 } 458 459 void CudaToolChain::addClangTargetOptions( 460 const llvm::opt::ArgList &DriverArgs, 461 llvm::opt::ArgStringList &CC1Args, 462 Action::OffloadKind DeviceOffloadingKind) const { 463 HostTC.addClangTargetOptions(DriverArgs, CC1Args, DeviceOffloadingKind); 464 465 StringRef GpuArch = DriverArgs.getLastArgValue(options::OPT_march_EQ); 466 assert(!GpuArch.empty() && "Must have an explicit GPU arch."); 467 assert((DeviceOffloadingKind == Action::OFK_OpenMP || 468 DeviceOffloadingKind == Action::OFK_Cuda) && 469 "Only OpenMP or CUDA offloading kinds are supported for NVIDIA GPUs."); 470 471 if (DeviceOffloadingKind == Action::OFK_Cuda) { 472 CC1Args.push_back("-fcuda-is-device"); 473 474 if (DriverArgs.hasFlag(options::OPT_fcuda_flush_denormals_to_zero, 475 options::OPT_fno_cuda_flush_denormals_to_zero, false)) 476 CC1Args.push_back("-fcuda-flush-denormals-to-zero"); 477 478 if (DriverArgs.hasFlag(options::OPT_fcuda_approx_transcendentals, 479 options::OPT_fno_cuda_approx_transcendentals, false)) 480 CC1Args.push_back("-fcuda-approx-transcendentals"); 481 482 if (DriverArgs.hasArg(options::OPT_nocudalib)) 483 return; 484 } 485 486 std::string LibDeviceFile = CudaInstallation.getLibDeviceFile(GpuArch); 487 488 if (LibDeviceFile.empty()) { 489 getDriver().Diag(diag::err_drv_no_cuda_libdevice) << GpuArch; 490 return; 491 } 492 493 CC1Args.push_back("-mlink-cuda-bitcode"); 494 CC1Args.push_back(DriverArgs.MakeArgString(LibDeviceFile)); 495 496 // Libdevice in CUDA-7.0 requires PTX version that's more recent 497 // than LLVM defaults to. Use PTX4.2 which is the PTX version that 498 // came with CUDA-7.0. 499 CC1Args.push_back("-target-feature"); 500 CC1Args.push_back("+ptx42"); 501 } 502 503 void CudaToolChain::AddCudaIncludeArgs(const ArgList &DriverArgs, 504 ArgStringList &CC1Args) const { 505 // Check our CUDA version if we're going to include the CUDA headers. 506 if (!DriverArgs.hasArg(options::OPT_nocudainc) && 507 !DriverArgs.hasArg(options::OPT_no_cuda_version_check)) { 508 StringRef Arch = DriverArgs.getLastArgValue(options::OPT_march_EQ); 509 assert(!Arch.empty() && "Must have an explicit GPU arch."); 510 CudaInstallation.CheckCudaVersionSupportsArch(StringToCudaArch(Arch)); 511 } 512 CudaInstallation.AddCudaIncludeArgs(DriverArgs, CC1Args); 513 } 514 515 llvm::opt::DerivedArgList * 516 CudaToolChain::TranslateArgs(const llvm::opt::DerivedArgList &Args, 517 StringRef BoundArch, 518 Action::OffloadKind DeviceOffloadKind) const { 519 DerivedArgList *DAL = 520 HostTC.TranslateArgs(Args, BoundArch, DeviceOffloadKind); 521 if (!DAL) 522 DAL = new DerivedArgList(Args.getBaseArgs()); 523 524 const OptTable &Opts = getDriver().getOpts(); 525 526 // For OpenMP device offloading, append derived arguments. Make sure 527 // flags are not duplicated. 528 // Also append the compute capability. 529 if (DeviceOffloadKind == Action::OFK_OpenMP) { 530 for (Arg *A : Args){ 531 bool IsDuplicate = false; 532 for (Arg *DALArg : *DAL){ 533 if (A == DALArg) { 534 IsDuplicate = true; 535 break; 536 } 537 } 538 if (!IsDuplicate) 539 DAL->append(A); 540 } 541 542 StringRef Arch = DAL->getLastArgValue(options::OPT_march_EQ); 543 if (Arch.empty()) { 544 // Default compute capability for CUDA toolchain is the 545 // lowest compute capability supported by the installed 546 // CUDA version. 547 DAL->AddJoinedArg(nullptr, 548 Opts.getOption(options::OPT_march_EQ), 549 CudaInstallation.getLowestExistingArch()); 550 } 551 552 return DAL; 553 } 554 555 for (Arg *A : Args) { 556 if (A->getOption().matches(options::OPT_Xarch__)) { 557 // Skip this argument unless the architecture matches BoundArch 558 if (BoundArch.empty() || A->getValue(0) != BoundArch) 559 continue; 560 561 unsigned Index = Args.getBaseArgs().MakeIndex(A->getValue(1)); 562 unsigned Prev = Index; 563 std::unique_ptr<Arg> XarchArg(Opts.ParseOneArg(Args, Index)); 564 565 // If the argument parsing failed or more than one argument was 566 // consumed, the -Xarch_ argument's parameter tried to consume 567 // extra arguments. Emit an error and ignore. 568 // 569 // We also want to disallow any options which would alter the 570 // driver behavior; that isn't going to work in our model. We 571 // use isDriverOption() as an approximation, although things 572 // like -O4 are going to slip through. 573 if (!XarchArg || Index > Prev + 1) { 574 getDriver().Diag(diag::err_drv_invalid_Xarch_argument_with_args) 575 << A->getAsString(Args); 576 continue; 577 } else if (XarchArg->getOption().hasFlag(options::DriverOption)) { 578 getDriver().Diag(diag::err_drv_invalid_Xarch_argument_isdriver) 579 << A->getAsString(Args); 580 continue; 581 } 582 XarchArg->setBaseArg(A); 583 A = XarchArg.release(); 584 DAL->AddSynthesizedArg(A); 585 } 586 DAL->append(A); 587 } 588 589 if (!BoundArch.empty()) { 590 DAL->eraseArg(options::OPT_march_EQ); 591 DAL->AddJoinedArg(nullptr, Opts.getOption(options::OPT_march_EQ), BoundArch); 592 } 593 return DAL; 594 } 595 596 Tool *CudaToolChain::buildAssembler() const { 597 return new tools::NVPTX::Assembler(*this); 598 } 599 600 Tool *CudaToolChain::buildLinker() const { 601 if (OK == Action::OFK_OpenMP) 602 return new tools::NVPTX::OpenMPLinker(*this); 603 return new tools::NVPTX::Linker(*this); 604 } 605 606 void CudaToolChain::addClangWarningOptions(ArgStringList &CC1Args) const { 607 HostTC.addClangWarningOptions(CC1Args); 608 } 609 610 ToolChain::CXXStdlibType 611 CudaToolChain::GetCXXStdlibType(const ArgList &Args) const { 612 return HostTC.GetCXXStdlibType(Args); 613 } 614 615 void CudaToolChain::AddClangSystemIncludeArgs(const ArgList &DriverArgs, 616 ArgStringList &CC1Args) const { 617 HostTC.AddClangSystemIncludeArgs(DriverArgs, CC1Args); 618 } 619 620 void CudaToolChain::AddClangCXXStdlibIncludeArgs(const ArgList &Args, 621 ArgStringList &CC1Args) const { 622 HostTC.AddClangCXXStdlibIncludeArgs(Args, CC1Args); 623 } 624 625 void CudaToolChain::AddIAMCUIncludeArgs(const ArgList &Args, 626 ArgStringList &CC1Args) const { 627 HostTC.AddIAMCUIncludeArgs(Args, CC1Args); 628 } 629 630 SanitizerMask CudaToolChain::getSupportedSanitizers() const { 631 // The CudaToolChain only supports sanitizers in the sense that it allows 632 // sanitizer arguments on the command line if they are supported by the host 633 // toolchain. The CudaToolChain will actually ignore any command line 634 // arguments for any of these "supported" sanitizers. That means that no 635 // sanitization of device code is actually supported at this time. 636 // 637 // This behavior is necessary because the host and device toolchains 638 // invocations often share the command line, so the device toolchain must 639 // tolerate flags meant only for the host toolchain. 640 return HostTC.getSupportedSanitizers(); 641 } 642 643 VersionTuple CudaToolChain::computeMSVCVersion(const Driver *D, 644 const ArgList &Args) const { 645 return HostTC.computeMSVCVersion(D, Args); 646 } 647