[llvm] [SPIR-V] Fix lowering of declarations with hidden visibility (PR #185029)
Dmitry Sidorov via llvm-commits
llvm-commits at lists.llvm.org
Fri Mar 6 07:54:59 PST 2026
https://github.com/MrSidims created https://github.com/llvm/llvm-project/pull/185029
They should be translated to SPIR-V and have Import linkage (unless they are Interface variables).
Also add a test for protected visibility.
>From 90f9140ebdf715a0476e64fb00de4b83905236c3 Mon Sep 17 00:00:00 2001
From: Dmitry Sidorov <Dmitry.Sidorov at amd.com>
Date: Fri, 6 Mar 2026 16:36:38 +0100
Subject: [PATCH] [SPIR-V] Fix lowering of declarations with hidden visibility
They should be translated to SPIR-V and have Import linkage (unless
they are Interface variables).
Also add a test for protected visibility.
---
.../SPIRV/MCTargetDesc/SPIRVMCAsmInfo.cpp | 3 ++
llvm/lib/Target/SPIRV/SPIRVUtils.cpp | 16 ++++++++--
.../SPIRV/linkage/hidden-interface-vars.ll | 25 +++++++++++++++
.../SPIRV/linkage/hidden-visibility.ll | 32 +++++++++++++++++++
.../SPIRV/linkage/protected-visibility.ll | 26 +++++++++++++++
5 files changed, 100 insertions(+), 2 deletions(-)
create mode 100644 llvm/test/CodeGen/SPIRV/linkage/hidden-interface-vars.ll
create mode 100644 llvm/test/CodeGen/SPIRV/linkage/hidden-visibility.ll
create mode 100644 llvm/test/CodeGen/SPIRV/linkage/protected-visibility.ll
diff --git a/llvm/lib/Target/SPIRV/MCTargetDesc/SPIRVMCAsmInfo.cpp b/llvm/lib/Target/SPIRV/MCTargetDesc/SPIRVMCAsmInfo.cpp
index 800bf2297fa7b..3f0a4c2eb5c76 100644
--- a/llvm/lib/Target/SPIRV/MCTargetDesc/SPIRVMCAsmInfo.cpp
+++ b/llvm/lib/Target/SPIRV/MCTargetDesc/SPIRVMCAsmInfo.cpp
@@ -27,6 +27,9 @@ SPIRVMCAsmInfo::SPIRVMCAsmInfo(const Triple &TT,
CodePointerSize = 4;
CommentString = ";";
HasFunctionAlignment = false;
+
+ HiddenDeclarationVisibilityAttr = HiddenVisibilityAttr = MCSA_Invalid;
+ ProtectedVisibilityAttr = MCSA_Invalid;
}
bool SPIRVMCAsmInfo::shouldOmitSectionDirective(StringRef SectionName) const {
diff --git a/llvm/lib/Target/SPIRV/SPIRVUtils.cpp b/llvm/lib/Target/SPIRV/SPIRVUtils.cpp
index 7858f7f932a46..4a49bbd1164bf 100644
--- a/llvm/lib/Target/SPIRV/SPIRVUtils.cpp
+++ b/llvm/lib/Target/SPIRV/SPIRVUtils.cpp
@@ -1197,11 +1197,23 @@ Type *reconstitutePeeledArrayType(Type *Ty) {
std::optional<SPIRV::LinkageType::LinkageType>
getSpirvLinkageTypeFor(const SPIRVSubtarget &ST, const GlobalValue &GV) {
- if (GV.hasLocalLinkage() || GV.hasHiddenVisibility())
+ if (GV.hasLocalLinkage())
return std::nullopt;
- if (GV.isDeclarationForLinker())
+ if (GV.isDeclarationForLinker()) {
+ // Interface variables mustn't not get Import linkage.
+ if (const auto *GVar = dyn_cast<GlobalVariable>(&GV)) {
+ auto SC = addressSpaceToStorageClass(GVar->getAddressSpace(), ST);
+ if (SC == SPIRV::StorageClass::Input ||
+ SC == SPIRV::StorageClass::Output ||
+ SC == SPIRV::StorageClass::PushConstant)
+ return std::nullopt;
+ }
return SPIRV::LinkageType::Import;
+ }
+
+ if (GV.hasHiddenVisibility())
+ return std::nullopt;
if (GV.hasLinkOnceODRLinkage() &&
ST.canUseExtension(SPIRV::Extension::SPV_KHR_linkonce_odr))
diff --git a/llvm/test/CodeGen/SPIRV/linkage/hidden-interface-vars.ll b/llvm/test/CodeGen/SPIRV/linkage/hidden-interface-vars.ll
new file mode 100644
index 0000000000000..29fa5cdf41ac1
--- /dev/null
+++ b/llvm/test/CodeGen/SPIRV/linkage/hidden-interface-vars.ll
@@ -0,0 +1,25 @@
+; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv-unknown-vulkan1.3-pixel %s -o - | FileCheck %s
+; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-unknown-vulkan1.3-pixel %s -o - -filetype=obj | spirv-val --target-env vulkan1.3 %}
+
+; Check that interface variables (Input, Output, PushConstant) do not get
+; Import linkage even when declared as external hidden.
+
+; CHECK-NOT: OpCapability Linkage
+; CHECK-NOT: LinkageAttributes
+
+ at input_var = external hidden addrspace(7) global <4 x float>
+ at output_var = external hidden addrspace(8) global <4 x float>
+
+; CHECK-DAG: %[[#INPUT_TYPE:]] = OpTypePointer Input %[[#FLOAT4:]]
+; CHECK-DAG: %[[#OUTPUT_TYPE:]] = OpTypePointer Output %[[#FLOAT4]]
+; CHECK-DAG: %[[#INPUT_VAR:]] = OpVariable %[[#INPUT_TYPE]] Input
+; CHECK-DAG: %[[#OUTPUT_VAR:]] = OpVariable %[[#OUTPUT_TYPE]] Output
+
+define void @main() #0 {
+entry:
+ %val = load <4 x float>, ptr addrspace(7) @input_var
+ store <4 x float> %val, ptr addrspace(8) @output_var
+ ret void
+}
+
+attributes #0 = { "hlsl.shader"="pixel" }
diff --git a/llvm/test/CodeGen/SPIRV/linkage/hidden-visibility.ll b/llvm/test/CodeGen/SPIRV/linkage/hidden-visibility.ll
new file mode 100644
index 0000000000000..de3ef5dc3277f
--- /dev/null
+++ b/llvm/test/CodeGen/SPIRV/linkage/hidden-visibility.ll
@@ -0,0 +1,32 @@
+; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s
+; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
+
+; Check that hidden visibility does not cause a crash and that hidden
+; declarations get Import linkage while hidden definitions do not.
+
+; CHECK-DAG: OpName %[[#HIDDEN_HELPER:]] "hidden_helper"
+; CHECK-DAG: OpName %[[#HIDDEN_DEF:]] "hidden_def"
+; CHECK-DAG: OpName %[[#HIDDEN_VAR:]] "hidden_leaf_var"
+; CHECK-DAG: OpName %[[#KERN:]] "test_kernel"
+
+; CHECK-DAG: OpDecorate %[[#HIDDEN_HELPER]] LinkageAttributes "hidden_helper" Import
+; CHECK-DAG: OpDecorate %[[#HIDDEN_VAR]] LinkageAttributes "hidden_leaf_var" Import
+; CHECK-NOT: OpDecorate %[[#HIDDEN_DEF]] LinkageAttributes
+
+ at hidden_leaf_var = external hidden addrspace(1) global i32
+
+declare hidden spir_func void @hidden_helper(ptr addrspace(1))
+
+define hidden spir_func void @hidden_def(ptr addrspace(1) %x) {
+entry:
+ ret void
+}
+
+define spir_kernel void @test_kernel(ptr addrspace(1) %data) {
+entry:
+ %val = load i32, ptr addrspace(1) @hidden_leaf_var
+ store i32 %val, ptr addrspace(1) %data
+ call spir_func void @hidden_helper(ptr addrspace(1) %data)
+ call spir_func void @hidden_def(ptr addrspace(1) %data)
+ ret void
+}
diff --git a/llvm/test/CodeGen/SPIRV/linkage/protected-visibility.ll b/llvm/test/CodeGen/SPIRV/linkage/protected-visibility.ll
new file mode 100644
index 0000000000000..41e973ed269d6
--- /dev/null
+++ b/llvm/test/CodeGen/SPIRV/linkage/protected-visibility.ll
@@ -0,0 +1,26 @@
+; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s
+; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
+
+; Check that protected visibility does not cause a crash, that protected
+; declarations get Import linkage, and protected definitions get Export.
+
+; CHECK-DAG: OpName %[[#PROTECTED_DECL:]] "protected_decl"
+; CHECK-DAG: OpName %[[#PROTECTED_DEF:]] "protected_def"
+; CHECK-DAG: OpName %[[#KERN:]] "test_kernel"
+
+; CHECK-DAG: OpDecorate %[[#PROTECTED_DECL]] LinkageAttributes "protected_decl" Import
+; CHECK-DAG: OpDecorate %[[#PROTECTED_DEF]] LinkageAttributes "protected_def" Export
+
+declare protected spir_func void @protected_decl(ptr addrspace(1))
+
+define protected spir_func void @protected_def(ptr addrspace(1) %x) {
+entry:
+ ret void
+}
+
+define spir_kernel void @test_kernel(ptr addrspace(1) %data) {
+entry:
+ call spir_func void @protected_decl(ptr addrspace(1) %data)
+ call spir_func void @protected_def(ptr addrspace(1) %data)
+ ret void
+}
More information about the llvm-commits
mailing list