1//===--- AMDGPU.cpp - AMDGPU Helpers for Tools ----------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "AMDGPU.h"
10#include "clang/Basic/TargetID.h"
11#include "clang/Driver/Driver.h"
12#include "clang/Options/Options.h"
13#include "llvm/TargetParser/AMDGPUTargetParser.h"
14
15using namespace clang::driver;
16using namespace clang::driver::tools;
17using namespace clang;
18using namespace llvm::opt;
19
20void AMDGPU::setArchNameInTriple(const Driver &D, const ArgList &Args,
21 BoundArch BA, types::ID InputType,
22 llvm::Triple &Triple) {
23 StringRef MArch = BA.ArchName;
24 if (MArch.empty())
25 AMDGPU::getAMDGPUArchCPUFromArgs(Triple, Args, Arch&: MArch);
26
27 if (MArch == "amdgcnspirv") {
28 Triple.setArch(Kind: llvm::Triple::ArchType::spirv64);
29 return;
30 }
31
32 StringRef ProcName = getProcessorFromTargetID(T: Triple, OffloadArch: MArch);
33 llvm::AMDGPU::GPUKind GK = llvm::AMDGPU::parseArchAMDGCN(CPU: ProcName);
34 if (GK == llvm::AMDGPU::GPUKind::GK_NONE) {
35 // Normalize legacy "amdgcn" triples to "amdgpu"
36 Triple.setArch(Kind: Triple.getArch(), SubArch: Triple.getSubArch());
37 return;
38 }
39
40 Triple.setArch(Kind: Triple.getArch(), SubArch: static_cast<llvm::Triple::SubArchType>(
41 llvm::AMDGPU::getSubArch(AK: GK)));
42}
43
44void AMDGPU::getAMDGPUArchCPUFromArgs(const llvm::Triple &Triple,
45 const llvm::opt::ArgList &Args,
46 llvm::StringRef &Arch) {
47 if (const Arg *MCPU = Args.getLastArg(Ids: options::OPT_mcpu_EQ))
48 Arch = MCPU->getValue();
49}
50