dm: blk: Use IS_ENABLED() instead of #ifdefs in blk_post_probe

Use IS_ENABLED() instead of #ifdef in blk_post_probe function.

No functional change intended.

Reviewed-by: Simon Glass <sjg@chromium.org>
Signed-off-by: Ovidiu Panait <ovidiu.panait@windriver.com>
[trini: Fix thinko and use CONFIG_HAVE_BLOCK_DEVICE in IS_ENABLED()]
Signed-off-by: Tom Rini <trini@konsulko.com>

diff --git a/drivers/block/blk-uclass.c b/drivers/block/blk-uclass.c
index b19375cbc8da3466eef5e22a3d27952b2140500d..b46a1ac8d210938ac20adf92375238e584637f0d 100644 (file)
--- a/drivers/block/blk-uclass.c
+++ b/drivers/block/blk-uclass.c
@@ -644,11 +644,12 @@ int blk_unbind_all(int if_type)
 
 static int blk_post_probe(struct udevice *dev)
 {
-#if defined(CONFIG_PARTITIONS) && defined(CONFIG_HAVE_BLOCK_DEVICE)
-       struct blk_desc *desc = dev_get_uclass_platdata(dev);
+       if (IS_ENABLED(CONFIG_PARTITIONS) &&
+           IS_ENABLED(CONFIG_HAVE_BLOCK_DEVICE)) {
+               struct blk_desc *desc = dev_get_uclass_platdata(dev);
 
-       part_init(desc);
-#endif
+               part_init(desc);
+       }
 
        return 0;
 }