[llvm] let user select among alternative enabling capabilities (PR #81476)
Vyacheslav Levytskyy via llvm-commits
llvm-commits at lists.llvm.org
Mon Feb 12 05:02:48 PST 2024
https://github.com/VyacheslavLevytskyy created https://github.com/llvm/llvm-project/pull/81476
By SPIR-V specification: "If an instruction, enumerant, or other feature specifies multiple enabling capabilities, only one such
capability needs to be declared to use the feature."
However, one capability may be preferred over another. One important case is Shader capability that may not be supported by a backend, but always is inserted if "OpDecorate SpecId" is found, because Enabling Capabilities for the latter is the list Shader, Kernel, where Shader is always selected as the first option.
In this PR we address the problem by leaving default behaviour as is, but introducing a new "--avoid-spirv-capabilities" command line option to avoid selection of certain capabilities if there are other options.
>From 60df0cf0d7f238e9643504076b05947903b0b25b Mon Sep 17 00:00:00 2001
From: "Levytskyy, Vyacheslav" <vyacheslav.levytskyy at intel.com>
Date: Mon, 12 Feb 2024 04:58:04 -0800
Subject: [PATCH] let user select among alternative enabling capabilities
---
llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp | 46 ++++++++++++++++---
llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.h | 6 +--
.../CodeGen/SPIRV/transcoding/spec_const.ll | 6 +--
3 files changed, 45 insertions(+), 13 deletions(-)
diff --git a/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp b/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp
index a18aae1761c834..9f14ea5dbe19b3 100644
--- a/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp
+++ b/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp
@@ -35,6 +35,22 @@ static cl::opt<bool>
cl::desc("Dump MIR with SPIR-V dependencies info"),
cl::Optional, cl::init(false));
+static cl::list<SPIRV::Capability::Capability>
+ AvoidCapabilities("avoid-spirv-capabilities",
+ cl::desc("SPIR-V capabilities to avoid if there are "
+ "other options enabling a feature"),
+ cl::ZeroOrMore, cl::Hidden,
+ cl::values(clEnumValN(SPIRV::Capability::Shader, "Shader",
+ "SPIR-V Shader capability")));
+// Use sets instead of cl::list to check "if contains" condition
+struct AvoidCapabilitiesSet {
+ SmallSet<SPIRV::Capability::Capability, 4> S;
+ AvoidCapabilitiesSet() {
+ for (auto Cap : AvoidCapabilities)
+ S.insert(Cap);
+ }
+};
+
char llvm::SPIRVModuleAnalysis::ID = 0;
namespace llvm {
@@ -58,6 +74,8 @@ static SPIRV::Requirements
getSymbolicOperandRequirements(SPIRV::OperandCategory::OperandCategory Category,
unsigned i, const SPIRVSubtarget &ST,
SPIRV::RequirementHandler &Reqs) {
+ static AvoidCapabilitiesSet
+ AvoidCaps; // contains capabilities to avoid if there is another option
unsigned ReqMinVer = getSymbolicOperandMinVersion(Category, i);
unsigned ReqMaxVer = getSymbolicOperandMaxVersion(Category, i);
unsigned TargetVer = ST.getSPIRVVersion();
@@ -72,9 +90,26 @@ getSymbolicOperandRequirements(SPIRV::OperandCategory::OperandCategory Category,
return {false, {}, {}, 0, 0};
}
} else if (MinVerOK && MaxVerOK) {
- for (auto Cap : ReqCaps) { // Only need 1 of the capabilities to work.
+ if (ReqCaps.size() == 1) {
+ auto Cap = ReqCaps[0];
if (Reqs.isCapabilityAvailable(Cap))
return {true, {Cap}, {}, ReqMinVer, ReqMaxVer};
+ } else {
+ // By SPIR-V specification: "If an instruction, enumerant, or other
+ // feature specifies multiple enabling capabilities, only one such
+ // capability needs to be declared to use the feature." However, one
+ // capability may be preferred over another. We use command line
+ // argument(s) and AvoidCapabilities to avoid selection of certain
+ // capabilities if there are other options.
+ CapabilityList UseCaps;
+ for (auto Cap : ReqCaps)
+ if (Reqs.isCapabilityAvailable(Cap))
+ UseCaps.push_back(Cap);
+ for (size_t i = 0, Sz = UseCaps.size(); i < Sz; ++i) {
+ auto Cap = UseCaps[i];
+ if (i == Sz - 1 || !AvoidCaps.S.contains(Cap))
+ return {true, {Cap}, {}, ReqMinVer, ReqMaxVer};
+ }
}
}
// If there are no capabilities, or we can't satisfy the version or
@@ -432,16 +467,13 @@ void SPIRV::RequirementHandler::getAndAddRequirements(
addRequirements(getSymbolicOperandRequirements(Category, i, ST, *this));
}
-void SPIRV::RequirementHandler::pruneCapabilities(
+void SPIRV::RequirementHandler::recursiveAddCapabilities(
const CapabilityList &ToPrune) {
for (const auto &Cap : ToPrune) {
AllCaps.insert(Cap);
- auto FoundIndex = llvm::find(MinimalCaps, Cap);
- if (FoundIndex != MinimalCaps.end())
- MinimalCaps.erase(FoundIndex);
CapabilityList ImplicitDecls =
getSymbolicOperandCapabilities(OperandCategory::CapabilityOperand, Cap);
- pruneCapabilities(ImplicitDecls);
+ recursiveAddCapabilities(ImplicitDecls);
}
}
@@ -452,7 +484,7 @@ void SPIRV::RequirementHandler::addCapabilities(const CapabilityList &ToAdd) {
continue;
CapabilityList ImplicitDecls =
getSymbolicOperandCapabilities(OperandCategory::CapabilityOperand, Cap);
- pruneCapabilities(ImplicitDecls);
+ recursiveAddCapabilities(ImplicitDecls);
MinimalCaps.push_back(Cap);
}
}
diff --git a/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.h b/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.h
index b05526b06e7da7..708384fc55f525 100644
--- a/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.h
+++ b/llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.h
@@ -71,9 +71,9 @@ struct RequirementHandler {
SmallSet<Extension::Extension, 4> AllExtensions;
unsigned MinVersion; // 0 if no min version is defined.
unsigned MaxVersion; // 0 if no max version is defined.
- // Remove a list of capabilities from dedupedCaps and add them to AllCaps,
- // recursing through their implicitly declared capabilities too.
- void pruneCapabilities(const CapabilityList &ToPrune);
+ // Add capabilities to AllCaps, recursing through their implicitly declared
+ // capabilities too.
+ void recursiveAddCapabilities(const CapabilityList &ToPrune);
void initAvailableCapabilitiesForOpenCL(const SPIRVSubtarget &ST);
void initAvailableCapabilitiesForVulkan(const SPIRVSubtarget &ST);
diff --git a/llvm/test/CodeGen/SPIRV/transcoding/spec_const.ll b/llvm/test/CodeGen/SPIRV/transcoding/spec_const.ll
index c47dccb35e14d5..8ce76534c50db5 100644
--- a/llvm/test/CodeGen/SPIRV/transcoding/spec_const.ll
+++ b/llvm/test/CodeGen/SPIRV/transcoding/spec_const.ll
@@ -1,9 +1,9 @@
-; RUN: llc -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV
-; XFAIL: *
+; RUN: llc -O0 -mtriple=spirv32-unknown-unknown --avoid-spirv-capabilities=Shader %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV
+; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown --avoid-spirv-capabilities=Shader %s -o - -filetype=obj | spirv-val %}
+; CHECK-SPIRV-DAG: OpCapability Kernel
; CHECK-SPIRV-NOT: OpCapability Matrix
; CHECK-SPIRV-NOT: OpCapability Shader
-; CHECK-SPIRV: OpCapability Kernel
; CHECK-SPIRV-DAG: OpDecorate %[[#SC0:]] SpecId 0
; CHECK-SPIRV-DAG: OpDecorate %[[#SC1:]] SpecId 1
More information about the llvm-commits
mailing list