[llvm] WebAssembly: Remove threadlocal_address intrinsics when disabling TLS (PR #88209)
Matthias Braun via llvm-commits
llvm-commits at lists.llvm.org
Tue Apr 9 16:32:01 PDT 2024
https://github.com/MatzeB created https://github.com/llvm/llvm-project/pull/88209
None
>From 1894bcc63bd112122c2012de5c43870cddb9031b Mon Sep 17 00:00:00 2001
From: Matthias Braun <matze at braunis.de>
Date: Tue, 9 Apr 2024 15:46:54 -0700
Subject: [PATCH] WebAssembly: Remove threadlocal_address intrinsics when
disabling TLS
---
.../WebAssembly/WebAssemblyTargetMachine.cpp | 13 +++++++++
llvm/test/CodeGen/WebAssembly/strip-tls.ll | 28 +++++++++++++++++++
2 files changed, 41 insertions(+)
create mode 100644 llvm/test/CodeGen/WebAssembly/strip-tls.ll
diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
index 70685b2e3bb2de..27b0bb2501d4ed 100644
--- a/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
+++ b/llvm/lib/Target/WebAssembly/WebAssemblyTargetMachine.cpp
@@ -289,6 +289,19 @@ class CoalesceFeaturesAndStripAtomics final : public ModulePass {
bool stripThreadLocals(Module &M) {
bool Stripped = false;
+ for (auto &F : M) {
+ for (auto &B : F) {
+ for (auto &I : make_early_inc_range(B)) {
+ if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(&I)) {
+ if (II->getIntrinsicID() == Intrinsic::threadlocal_address) {
+ II->replaceAllUsesWith(II->getArgOperand(0));
+ II->eraseFromParent();
+ Stripped = true;
+ }
+ }
+ }
+ }
+ }
for (auto &GV : M.globals()) {
if (GV.isThreadLocal()) {
Stripped = true;
diff --git a/llvm/test/CodeGen/WebAssembly/strip-tls.ll b/llvm/test/CodeGen/WebAssembly/strip-tls.ll
new file mode 100644
index 00000000000000..281fe0aee7a497
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/strip-tls.ll
@@ -0,0 +1,28 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 4
+; RUN: llc -o - %s -mattr=-bulk-memory | FileCheck --check-prefix=NOTLS %s
+; RUN: llc -o - %s -mattr=+bulk-memory,+atomics | FileCheck --check-prefix=TLS %s
+
+target triple = "wasm32--"
+
+ at tls = internal thread_local global i32 0
+
+define i32 @func() {
+; NOTLS-LABEL: func:
+; NOTLS: .functype func () -> (i32)
+; NOTLS-NEXT: # %bb.0:
+; NOTLS-NEXT: i32.const 0
+; NOTLS-NEXT: i32.load tls
+; NOTLS-NEXT: # fallthrough-return
+;
+; TLS-LABEL: func:
+; TLS: .functype func () -> (i32)
+; TLS-NEXT: # %bb.0:
+; TLS-NEXT: global.get __tls_base
+; TLS-NEXT: i32.const tls at TLSREL
+; TLS-NEXT: i32.add
+; TLS-NEXT: i32.load 0
+; TLS-NEXT: # fallthrough-return
+ %p = call ptr @llvm.threadlocal.address.p0(ptr @tls)
+ %v = load i32, ptr %p
+ ret i32 %v
+}
More information about the llvm-commits
mailing list