Keep it together with the rest of the zoned code.

Signed-off-by: Christoph Hellwig <[email protected]>
Reviewed-by: Johannes Thumshirn <[email protected]>
---
 drivers/md/dm-table.c | 3 ---
 drivers/md/dm-zone.c  | 8 +++++++-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/drivers/md/dm-table.c b/drivers/md/dm-table.c
index cc66a27c363a65..e291b78b307b13 100644
--- a/drivers/md/dm-table.c
+++ b/drivers/md/dm-table.c
@@ -2040,9 +2040,6 @@ int dm_table_set_restrictions(struct dm_table *t, struct 
request_queue *q,
                r = dm_set_zones_restrictions(t, q);
                if (r)
                        return r;
-               if (blk_queue_is_zoned(q) &&
-                   !static_key_enabled(&zoned_enabled.key))
-                       static_branch_enable(&zoned_enabled);
        }
 
        dm_update_crypto_profile(q, t);
diff --git a/drivers/md/dm-zone.c b/drivers/md/dm-zone.c
index 8e6bcb0d786a1a..3103360ce7f040 100644
--- a/drivers/md/dm-zone.c
+++ b/drivers/md/dm-zone.c
@@ -287,7 +287,13 @@ int dm_set_zones_restrictions(struct dm_table *t, struct 
request_queue *q)
                       queue_emulates_zone_append(q) ? "emulated" : "native");
        }
 
-       return dm_revalidate_zones(md, t);
+       ret = dm_revalidate_zones(md, t);
+       if (ret < 0)
+               return ret;
+
+       if (!static_key_enabled(&zoned_enabled.key))
+               static_branch_enable(&zoned_enabled);
+       return 0;
 }
 
 /*
-- 
2.43.0


Reply via email to