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 234fa6eedbf78bea0394b42a061a6010b8a92ed3..63e426e82bd3933b979fd938d3adf17088c5d259 100644 (file)
--- 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));