13#include "llvm/ADT/StringSwitch.h"
14#include "llvm/TargetParser/Triple.h"
26const unsigned OfflineLibMinor = 0xF;
28bool isLegalShaderModel(Triple &
T) {
29 if (
T.getOS() != Triple::OSType::ShaderModel)
32 auto Version =
T.getOSVersion();
33 if (Version.getBuild())
35 if (Version.getSubminor())
38 auto Kind =
T.getEnvironment();
43 case Triple::EnvironmentType::Vertex:
44 case Triple::EnvironmentType::Hull:
45 case Triple::EnvironmentType::Domain:
46 case Triple::EnvironmentType::Geometry:
47 case Triple::EnvironmentType::Pixel:
48 case Triple::EnvironmentType::Compute: {
49 VersionTuple MinVer(4, 0);
50 return MinVer <= Version;
52 case Triple::EnvironmentType::Library: {
53 VersionTuple SM6x(6, OfflineLibMinor);
57 VersionTuple MinVer(6, 3);
58 return MinVer <= Version;
60 case Triple::EnvironmentType::Amplification:
61 case Triple::EnvironmentType::Mesh: {
62 VersionTuple MinVer(6, 5);
63 return MinVer <= Version;
65 case Triple::EnvironmentType::RootSignature:
66 VersionTuple MinVer(1, 0);
67 VersionTuple MaxVer(1, 2);
68 return MinVer <= Version && Version <= MaxVer;
73std::optional<llvm::Triple> tryParseTriple(StringRef Profile) {
76 Profile.split(Parts,
"_");
77 if (Parts.size() != 3)
80 Triple::EnvironmentType Kind =
81 StringSwitch<Triple::EnvironmentType>(Parts[0])
82 .Case(
"ps", Triple::EnvironmentType::Pixel)
83 .Case(
"vs", Triple::EnvironmentType::Vertex)
84 .Case(
"gs", Triple::EnvironmentType::Geometry)
85 .Case(
"hs", Triple::EnvironmentType::Hull)
86 .Case(
"ds", Triple::EnvironmentType::Domain)
87 .Case(
"cs", Triple::EnvironmentType::Compute)
88 .Case(
"lib", Triple::EnvironmentType::Library)
89 .Case(
"ms", Triple::EnvironmentType::Mesh)
90 .Case(
"as", Triple::EnvironmentType::Amplification)
91 .Case(
"rootsig", Triple::EnvironmentType::RootSignature)
92 .Default(Triple::EnvironmentType::UnknownEnvironment);
93 if (Kind == Triple::EnvironmentType::UnknownEnvironment)
96 unsigned long long Major = 0;
97 if (llvm::getAsUnsignedInteger(Parts[1], 0, Major))
100 unsigned long long Minor = 0;
101 if (Parts[2] ==
"x" && Kind == Triple::EnvironmentType::Library)
102 Minor = OfflineLibMinor;
103 else if (llvm::getAsUnsignedInteger(Parts[2], 0, Minor))
111 Triple::SubArchType SubArch = llvm::Triple::NoSubArch;
114 SubArch = llvm::Triple::DXILSubArch_v1_0;
117 SubArch = llvm::Triple::DXILSubArch_v1_1;
120 SubArch = llvm::Triple::DXILSubArch_v1_2;
123 SubArch = llvm::Triple::DXILSubArch_v1_3;
126 SubArch = llvm::Triple::DXILSubArch_v1_4;
129 SubArch = llvm::Triple::DXILSubArch_v1_5;
132 SubArch = llvm::Triple::DXILSubArch_v1_6;
135 SubArch = llvm::Triple::DXILSubArch_v1_7;
138 SubArch = llvm::Triple::DXILSubArch_v1_8;
141 SubArch = llvm::Triple::DXILSubArch_v1_9;
143 case OfflineLibMinor:
145 SubArch = llvm::Triple::LatestDXILSubArch;
151 T.setArch(Triple::ArchType::dxil, SubArch);
152 T.setOSName(Triple::getOSTypeName(Triple::OSType::ShaderModel).str() +
154 T.setEnvironment(Kind);
159std::optional<std::string> tryParseProfile(StringRef Profile) {
160 std::optional<llvm::Triple> MaybeT = tryParseTriple(Profile);
161 if (MaybeT && isLegalShaderModel(*MaybeT))
162 return MaybeT->getTriple();
167bool isLegalValidatorVersion(StringRef ValVersionStr,
const Driver &D) {
168 VersionTuple Version;
169 if (Version.tryParse(ValVersionStr) || Version.getBuild() ||
170 Version.getSubminor() || !Version.getMinor()) {
171 D.
Diag(diag::err_drv_invalid_format_dxil_validator_version)
176 uint64_t Major = Version.getMajor();
177 uint64_t Minor = *Version.getMinor();
178 if (Major == 0 && Minor != 0) {
179 D.
Diag(diag::err_drv_invalid_empty_dxil_validator_version) << ValVersionStr;
182 VersionTuple MinVer(1, 0);
183 if (Version < MinVer) {
184 D.
Diag(diag::err_drv_invalid_range_dxil_validator_version) << ValVersionStr;
190void getSpirvExtOperand(StringRef SpvExtensionArg, raw_ostream &
out) {
193 static const std::vector<StringRef> DxcSupportedExtensions = {
194 "SPV_KHR_16bit_storage",
195 "SPV_KHR_device_group",
196 "SPV_KHR_fragment_shading_rate",
198 "SPV_KHR_post_depth_coverage",
199 "SPV_KHR_non_semantic_info",
200 "SPV_KHR_shader_draw_parameters",
201 "SPV_KHR_ray_tracing",
202 "SPV_KHR_shader_clock",
203 "SPV_EXT_demote_to_helper_invocation",
204 "SPV_EXT_descriptor_indexing",
205 "SPV_EXT_fragment_fully_covered",
206 "SPV_EXT_fragment_invocation_density",
207 "SPV_EXT_fragment_shader_interlock",
208 "SPV_EXT_mesh_shader",
209 "SPV_EXT_shader_stencil_export",
210 "SPV_EXT_shader_viewport_index_layer",
212 "SPV_GOOGLE_hlsl_functionality1",
213 "SPV_GOOGLE_user_type",
215 "SPV_EXT_shader_image_int64",
216 "SPV_KHR_fragment_shader_barycentric",
217 "SPV_KHR_physical_storage_buffer",
218 "SPV_KHR_vulkan_memory_model",
220 "SPV_KHR_maximal_reconvergence",
221 "SPV_KHR_float_controls",
222 "SPV_NV_shader_subgroup_partitioned",
226 if (SpvExtensionArg.starts_with(
"SPV_")) {
227 out <<
"+" << SpvExtensionArg;
231 if (SpvExtensionArg.compare_insensitive(
"DXC") == 0) {
233 for (StringRef E : DxcSupportedExtensions) {
242 out << SpvExtensionArg;
246 if (SpvExtensionArgs.empty()) {
247 return StringRef(
"-spirv-ext=all");
251 raw_svector_ostream
out(LlvmOption);
253 out <<
"-spirv-ext=";
254 getSpirvExtOperand(SpvExtensionArgs[0],
out);
256 SpvExtensionArgs = SpvExtensionArgs.slice(1);
257 for (StringRef Extension : SpvExtensionArgs) {
259 getSpirvExtOperand(Extension,
out);
264bool isValidSPIRVExtensionName(
const std::string &str) {
265 std::regex pattern(
"dxc|DXC|khr|KHR|SPV_[a-zA-Z0-9_]+");
266 return std::regex_match(str, pattern);
273 bool AllValid =
true;
274 for (
auto Extension : SpvExtensionArgs) {
275 if (!isValidSPIRVExtensionName(Extension)) {
277 <<
"-fspv-extension" << Extension;
284bool isRootSignatureTarget(StringRef Profile) {
285 if (std::optional<llvm::Triple>
T = tryParseTriple(Profile))
286 return T->getEnvironment() == Triple::EnvironmentType::RootSignature;
290bool isRootSignatureTarget(DerivedArgList &Args) {
291 if (
const Arg *A = Args.getLastArg(options::OPT_target_profile))
292 return isRootSignatureTarget(A->getValue());
302 const char *LinkingOutput)
const {
304 assert(DxvPath !=
"dxv" &&
"cannot find dxv");
306 ArgStringList CmdArgs;
307 assert(Inputs.size() == 1 &&
"Unable to handle multiple inputs.");
310 CmdArgs.push_back(
"-o");
313 const char *Exec = Args.MakeArgString(DxvPath);
315 Exec, CmdArgs, Inputs, Input));
321 const char *LinkingOutput)
const {
322 std::string MSCPath =
getToolChain().GetProgramPath(
"metal-shaderconverter");
323 ArgStringList CmdArgs;
324 assert(Inputs.size() == 1 &&
"Unable to handle multiple inputs.");
327 CmdArgs.push_back(
"-o");
330 const char *Exec = Args.MakeArgString(MSCPath);
332 Exec, CmdArgs, Inputs, Input));
339 const char *LinkingOutput)
const {
341 std::string ObjcopyPath =
getToolChain().GetProgramPath(
"llvm-objcopy");
342 const char *Exec = Args.MakeArgString(ObjcopyPath);
344 ArgStringList CmdArgs;
345 assert(Inputs.size() == 1 &&
"Unable to handle multiple inputs.");
350 if (Args.hasArg(options::OPT_dxc_strip_rootsignature)) {
351 const char *StripRS = Args.MakeArgString(
"--remove-section=RTS0");
352 CmdArgs.push_back(StripRS);
355 if (Arg *Arg = Args.getLastArg(options::OPT_dxc_Frs)) {
357 Args.MakeArgString(
"--extract-section=RTS0=" + Twine(Arg->getValue()));
358 CmdArgs.push_back(Frs);
361 if (
const Arg *A = Args.getLastArg(options::OPT_target_profile))
362 if (isRootSignatureTarget(A->getValue())) {
363 const char *Fos = Args.MakeArgString(
"--only-section=RTS0");
364 CmdArgs.push_back(Fos);
367 assert(CmdArgs.size() > 2 &&
"Unnecessary invocation of objcopy.");
370 Exec, CmdArgs, Inputs, Input));
377 if (Args.hasArg(options::OPT_dxc_validator_path_EQ))
379 Args.getLastArgValue(options::OPT_dxc_validator_path_EQ).str());
388 return Validator.get();
392 return MetalConverter.get();
396 return LLVMObjcopy.get();
402std::optional<std::string>
404 StringRef TargetProfile) {
405 return tryParseProfile(TargetProfile);
411 DerivedArgList *DAL =
new DerivedArgList(Args.getBaseArgs());
415 for (Arg *A : Args) {
416 if (A->getOption().getID() == options::OPT_dxil_validator_version) {
417 StringRef ValVerStr = A->getValue();
418 if (!isLegalValidatorVersion(ValVerStr,
getDriver()))
421 if (A->getOption().getID() == options::OPT_dxc_entrypoint) {
422 DAL->AddSeparateArg(
nullptr, Opts.getOption(options::OPT_hlsl_entrypoint),
427 if (A->getOption().getID() == options::OPT_dxc_rootsig_ver) {
428 DAL->AddJoinedArg(
nullptr,
429 Opts.getOption(options::OPT_fdx_rootsignature_version),
434 if (A->getOption().getID() == options::OPT_dxc_rootsig_define) {
435 DAL->AddJoinedArg(
nullptr,
436 Opts.getOption(options::OPT_fdx_rootsignature_define),
441 if (A->getOption().getID() == options::OPT__SLASH_O) {
442 StringRef OStr = A->getValue();
444 DAL->AddFlagArg(
nullptr, Opts.getOption(options::OPT_O0));
448 DAL->AddJoinedArg(
nullptr, Opts.getOption(options::OPT_O), OStr);
453 if (A->getOption().getID() == options::OPT_emit_pristine_llvm) {
455 DAL->AddFlagArg(
nullptr, Opts.getOption(options::OPT_emit_llvm));
456 DAL->AddFlagArg(
nullptr,
457 Opts.getOption(options::OPT_disable_llvm_passes));
461 if (A->getOption().getID() == options::OPT_dxc_hlsl_version) {
467 DAL->AddSeparateArg(
nullptr, Opts.getOption(options::OPT_std_EQ),
470 getDriver().
Diag(diag::err_drv_invalid_value) <<
"HV" << A->getValue();
476 if (A->getOption().getID() == options::OPT_dxc_gis) {
478 DAL->AddSeparateArg(
nullptr, Opts.getOption(options::OPT_ffp_model_EQ),
483 if (A->getOption().getID() == options::OPT_fvk_use_dx_layout) {
489 if (A->getOption().getID() == options::OPT_fvk_use_scalar_layout) {
490 getDriver().
Diag(diag::err_drv_clang_unsupported) << A->getAsString(Args);
495 if (A->getOption().getID() == options::OPT_fvk_use_gl_layout) {
496 getDriver().
Diag(diag::err_drv_clang_unsupported) << A->getAsString(Args);
504 if (
getArch() == llvm::Triple::spirv) {
505 std::vector<std::string> SpvExtensionArgs =
506 Args.getAllArgValues(options::OPT_fspv_extension_EQ);
507 if (checkExtensionArgsAreValid(SpvExtensionArgs,
getDriver())) {
509 DAL->AddSeparateArg(
nullptr, Opts.getOption(options::OPT_mllvm),
512 Args.claimAllArgs(options::OPT_fspv_extension_EQ);
515 if (!DAL->hasArg(options::OPT_O_Group)) {
516 DAL->AddJoinedArg(
nullptr, Opts.getOption(options::OPT_O),
"3");
523 if (!Args.hasArg(options::OPT_dxc_Fo))
526 if (Args.getLastArg(options::OPT_dxc_disable_validation))
530 if (DxvPath !=
"dxv")
538 return Args.hasArg(options::OPT_metal) && Args.hasArg(options::OPT_dxc_Fo);
542 return Args.hasArg(options::OPT_dxc_Fo) &&
543 (Args.hasArg(options::OPT_dxc_strip_rootsignature) ||
544 Args.hasArg(options::OPT_dxc_Frs) || isRootSignatureTarget(Args));
551 return AC == Action::Action::BinaryTranslatorJobClass;
553 return AC == Action::Action::BinaryAnalyzeJobClass;
555 return AC == Action::Action::ObjcopyJobClass;
@ BinaryTranslatorJobClass
Compilation - A set of tasks to perform for a single driver invocation.
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
DiagnosticBuilder Diag(unsigned DiagID) const
const llvm::opt::OptTable & getOpts() const
SmallVector< InputInfo, 4 > InputInfoList
The JSON file list parser is used to communicate input to InstallAPI.
const FunctionProtoType * T
llvm::StringRef getAsString(SyncScope S)
Diagnostic wrappers for TextAPI types for error reporting.
LangStandard - Information about the properties of a particular language standard.
static Kind getHLSLLangKind(StringRef Name)
static const LangStandard & getLangStandardForKind(Kind K)
const char * getName() const
getName - Get the name of this standard.
static constexpr ResponseFileSupport None()
Returns a ResponseFileSupport indicating that response files are not supported.