a65771fce4
Prefer using `llvm-spirv-<LLVM_VERSION_MAJOR>` tool (i.e. `llvm-spirv-18`) over plain `llvm-spirv`. If the versioned tool is not found in PATH, fall back to use the plain `llvm-spirv`. An issue with the using `llvm-spirv` is that the one found in PATH might be compiled against older LLVM version which could lead to crashes or obscure bugs. For example, `llvm-spirv` distributed by Ubuntu links against different LLVM version depending on the Ubuntu release (LLVM-10 in 20.04LTS, LLVM-13 in 22.04LTS).
102 lines
3.9 KiB
C++
102 lines
3.9 KiB
C++
//===--- SPIRV.cpp - SPIR-V Tool Implementations ----------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "SPIRV.h"
|
|
#include "CommonArgs.h"
|
|
#include "clang/Basic/Version.h"
|
|
#include "clang/Driver/Compilation.h"
|
|
#include "clang/Driver/Driver.h"
|
|
#include "clang/Driver/InputInfo.h"
|
|
#include "clang/Driver/Options.h"
|
|
|
|
using namespace clang::driver;
|
|
using namespace clang::driver::toolchains;
|
|
using namespace clang::driver::tools;
|
|
using namespace llvm::opt;
|
|
|
|
void SPIRV::constructTranslateCommand(Compilation &C, const Tool &T,
|
|
const JobAction &JA,
|
|
const InputInfo &Output,
|
|
const InputInfo &Input,
|
|
const llvm::opt::ArgStringList &Args) {
|
|
llvm::opt::ArgStringList CmdArgs(Args);
|
|
CmdArgs.push_back(Input.getFilename());
|
|
|
|
if (Input.getType() == types::TY_PP_Asm)
|
|
CmdArgs.push_back("-to-binary");
|
|
if (Output.getType() == types::TY_PP_Asm)
|
|
CmdArgs.push_back("--spirv-tools-dis");
|
|
|
|
CmdArgs.append({"-o", Output.getFilename()});
|
|
|
|
// Try to find "llvm-spirv-<LLVM_VERSION_MAJOR>". Otherwise, fall back to
|
|
// plain "llvm-spirv".
|
|
using namespace std::string_literals;
|
|
auto VersionedTool = "llvm-spirv-"s + std::to_string(LLVM_VERSION_MAJOR);
|
|
std::string ExeCand = T.getToolChain().GetProgramPath(VersionedTool.c_str());
|
|
if (!llvm::sys::fs::can_execute(ExeCand))
|
|
ExeCand = T.getToolChain().GetProgramPath("llvm-spirv");
|
|
|
|
const char *Exec = C.getArgs().MakeArgString(ExeCand);
|
|
C.addCommand(std::make_unique<Command>(JA, T, ResponseFileSupport::None(),
|
|
Exec, CmdArgs, Input, Output));
|
|
}
|
|
|
|
void SPIRV::Translator::ConstructJob(Compilation &C, const JobAction &JA,
|
|
const InputInfo &Output,
|
|
const InputInfoList &Inputs,
|
|
const ArgList &Args,
|
|
const char *LinkingOutput) const {
|
|
claimNoWarnArgs(Args);
|
|
if (Inputs.size() != 1)
|
|
llvm_unreachable("Invalid number of input files.");
|
|
constructTranslateCommand(C, *this, JA, Output, Inputs[0], {});
|
|
}
|
|
|
|
clang::driver::Tool *SPIRVToolChain::getTranslator() const {
|
|
if (!Translator)
|
|
Translator = std::make_unique<SPIRV::Translator>(*this);
|
|
return Translator.get();
|
|
}
|
|
|
|
clang::driver::Tool *SPIRVToolChain::SelectTool(const JobAction &JA) const {
|
|
Action::ActionClass AC = JA.getKind();
|
|
return SPIRVToolChain::getTool(AC);
|
|
}
|
|
|
|
clang::driver::Tool *SPIRVToolChain::getTool(Action::ActionClass AC) const {
|
|
switch (AC) {
|
|
default:
|
|
break;
|
|
case Action::BackendJobClass:
|
|
case Action::AssembleJobClass:
|
|
return SPIRVToolChain::getTranslator();
|
|
}
|
|
return ToolChain::getTool(AC);
|
|
}
|
|
clang::driver::Tool *SPIRVToolChain::buildLinker() const {
|
|
return new tools::SPIRV::Linker(*this);
|
|
}
|
|
|
|
void SPIRV::Linker::ConstructJob(Compilation &C, const JobAction &JA,
|
|
const InputInfo &Output,
|
|
const InputInfoList &Inputs,
|
|
const ArgList &Args,
|
|
const char *LinkingOutput) const {
|
|
const ToolChain &ToolChain = getToolChain();
|
|
std::string Linker = ToolChain.GetProgramPath(getShortName());
|
|
ArgStringList CmdArgs;
|
|
AddLinkerInputs(getToolChain(), Inputs, Args, CmdArgs, JA);
|
|
|
|
CmdArgs.push_back("-o");
|
|
CmdArgs.push_back(Output.getFilename());
|
|
|
|
C.addCommand(std::make_unique<Command>(JA, *this, ResponseFileSupport::None(),
|
|
Args.MakeArgString(Linker), CmdArgs,
|
|
Inputs, Output));
|
|
}
|