1 files changed, 2 insertions, 14 deletions

diff --git a/drivers/nvmem/mtk-efuse.c b/drivers/nvmem/mtk-efuse.c
index 9ee3479cfc7b..e66adf17a747 100644
--- a/drivers/nvmem/mtk-efuse.c
+++ b/drivers/nvmem/mtk-efuse.c
@@ -72,20 +72,9 @@ static int mtk_efuse_probe(struct platform_device *pdev)
         econfig.size = resource_size(res);
         econfig.priv = priv;
         econfig.dev = dev;
-        nvmem = nvmem_register(&econfig);
-        if (IS_ERR(nvmem))
-                return PTR_ERR(nvmem);
+        nvmem = devm_nvmem_register(dev, &econfig);
 
-        platform_set_drvdata(pdev, nvmem);
-
-        return 0;
-}
-
-static int mtk_efuse_remove(struct platform_device *pdev)
-{
-        struct nvmem_device *nvmem = platform_get_drvdata(pdev);
-
-        return nvmem_unregister(nvmem);
+        return PTR_ERR_OR_ZERO(nvmem);
 }
 
 static const struct of_device_id mtk_efuse_of_match[] = {
@@ -97,7 +86,6 @@ MODULE_DEVICE_TABLE(of, mtk_efuse_of_match);
 
 static struct platform_driver mtk_efuse_driver = {
         .probe = mtk_efuse_probe,
-        .remove = mtk_efuse_remove,
         .driver = {
                 .name = "mediatek,efuse",
                 .of_match_table = mtk_efuse_of_match,