From: Sidhartha Kumar <sidhartha.kumar@oracle.com>
Date: Thu, 18 Jan 2024 19:07:35 +0000 (-0800)
Subject: maple_tree: set store type in mas_store_prealloc()
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=00b97c1ee9b4a558b91575c610be8fa0516071d9;p=users%2Fjedix%2Flinux-maple.git

maple_tree: set store type in mas_store_prealloc()

Set the store type in mas_store_prealloc(). This is needed as
mas_wr_store_entry() will be modified to read the store type to make a
decision about how to complete the store.

Signed-off-by: Sidhartha Kumar <sidhartha.kumar@oracle.com>
---

diff --git a/lib/maple_tree.c b/lib/maple_tree.c
index 234fa6eedbf7..63e426e82bd3 100644
--- a/lib/maple_tree.c
+++ b/lib/maple_tree.c
@@ -5583,7 +5583,8 @@ void mas_store_prealloc(struct ma_state *mas, void *entry)
 {
 	MA_WR_STATE(wr_mas, mas, entry);
 
-	mas_wr_store_setup(&wr_mas);
+	mas_wr_prealloc_setup(&wr_mas);
+	mas_wr_store_type(&wr_mas);
 	trace_ma_write(__func__, mas, 0, entry);
 	mas_wr_store_entry(&wr_mas);
 	MAS_WR_BUG_ON(&wr_mas, mas_is_err(mas));