.name           = "nvmem",
 };
 
-static struct nvmem_device *of_nvmem_find(struct device_node *nvmem_np)
-{
-       struct device *d;
-
-       if (!nvmem_np)
-               return NULL;
-
-       d = bus_find_device_by_of_node(&nvmem_bus_type, nvmem_np);
-
-       if (!d)
-               return NULL;
-
-       return to_nvmem_device(d);
-}
-
-static struct nvmem_device *nvmem_find(const char *name)
-{
-       struct device *d;
-
-       d = bus_find_device_by_name(&nvmem_bus_type, NULL, name);
-
-       if (!d)
-               return NULL;
-
-       return to_nvmem_device(d);
-}
-
 static void nvmem_cell_drop(struct nvmem_cell *cell)
 {
        blocking_notifier_call_chain(&nvmem_notifier, NVMEM_CELL_REMOVE, cell);
 }
 EXPORT_SYMBOL(devm_nvmem_unregister);
 
-static struct nvmem_device *__nvmem_device_get(struct device_node *np,
-                                              const char *nvmem_name)
+static struct nvmem_device *__nvmem_device_get(void *data,
+                       int (*match)(struct device *dev, const void *data))
 {
        struct nvmem_device *nvmem = NULL;
+       struct device *dev;
 
        mutex_lock(&nvmem_mutex);
-       nvmem = np ? of_nvmem_find(np) : nvmem_find(nvmem_name);
+       dev = bus_find_device(&nvmem_bus_type, NULL, data, match);
+       if (dev)
+               nvmem = to_nvmem_device(dev);
        mutex_unlock(&nvmem_mutex);
        if (!nvmem)
                return ERR_PTR(-EPROBE_DEFER);
        if (!nvmem_np)
                return ERR_PTR(-ENOENT);
 
-       return __nvmem_device_get(nvmem_np, NULL);
+       return __nvmem_device_get(nvmem_np, device_match_of_node);
 }
 EXPORT_SYMBOL_GPL(of_nvmem_device_get);
 #endif
 
        }
 
-       return __nvmem_device_get(NULL, dev_name);
+       return __nvmem_device_get((void *)dev_name, device_match_name);
 }
 EXPORT_SYMBOL_GPL(nvmem_device_get);
 
+/**
+ * nvmem_device_find() - Find nvmem device with matching function
+ *
+ * @data: Data to pass to match function
+ * @match: Callback function to check device
+ *
+ * Return: ERR_PTR() on error or a valid pointer to a struct nvmem_device
+ * on success.
+ */
+struct nvmem_device *nvmem_device_find(void *data,
+                       int (*match)(struct device *dev, const void *data))
+{
+       return __nvmem_device_get(data, match);
+}
+EXPORT_SYMBOL_GPL(nvmem_device_find);
+
 static int devm_nvmem_device_match(struct device *dev, void *res, void *data)
 {
        struct nvmem_device **nvmem = res;
                if ((strcmp(lookup->dev_id, dev_id) == 0) &&
                    (strcmp(lookup->con_id, con_id) == 0)) {
                        /* This is the right entry. */
-                       nvmem = __nvmem_device_get(NULL, lookup->nvmem_name);
+                       nvmem = __nvmem_device_get((void *)lookup->nvmem_name,
+                                                  device_match_name);
                        if (IS_ERR(nvmem)) {
                                /* Provider may not be registered yet. */
                                cell = ERR_CAST(nvmem);
        if (!nvmem_np)
                return ERR_PTR(-EINVAL);
 
-       nvmem = __nvmem_device_get(nvmem_np, NULL);
+       nvmem = __nvmem_device_get(nvmem_np, device_match_of_node);
        of_node_put(nvmem_np);
        if (IS_ERR(nvmem))
                return ERR_CAST(nvmem);
 
 int nvmem_register_notifier(struct notifier_block *nb);
 int nvmem_unregister_notifier(struct notifier_block *nb);
 
+struct nvmem_device *nvmem_device_find(void *data,
+                       int (*match)(struct device *dev, const void *data));
+
 #else
 
 static inline struct nvmem_cell *nvmem_cell_get(struct device *dev,
        return -EOPNOTSUPP;
 }
 
+static inline struct nvmem_device *nvmem_device_find(void *data,
+                       int (*match)(struct device *dev, const void *data))
+{
+       return NULL;
+}
+
 #endif /* CONFIG_NVMEM */
 
 #if IS_ENABLED(CONFIG_NVMEM) && IS_ENABLED(CONFIG_OF)