/**
  * of_gpio_spi_cs_get_count() - special GPIO counting for SPI
- * @dev:    Consuming device
+ * @np:    Consuming device node
  * @con_id: Function within the GPIO consumer
  *
  * Some elder GPIO controllers need special quirks. Currently we handle
  * the counting of "cs-gpios" to count "gpios" transparent to the
  * driver.
  */
-static int of_gpio_spi_cs_get_count(struct device *dev, const char *con_id)
+static int of_gpio_spi_cs_get_count(const struct device_node *np,
+                                   const char *con_id)
 {
-       struct device_node *np = dev->of_node;
-
        if (!IS_ENABLED(CONFIG_SPI_MASTER))
                return 0;
        if (!con_id || strcmp(con_id, "cs"))
        return of_gpio_named_count(np, "gpios");
 }
 
-int of_gpio_get_count(struct device *dev, const char *con_id)
+int of_gpio_count(const struct fwnode_handle *fwnode, const char *con_id)
 {
+       const struct device_node *np = to_of_node(fwnode);
        int ret;
        char propname[32];
        unsigned int i;
 
-       ret = of_gpio_spi_cs_get_count(dev, con_id);
+       ret = of_gpio_spi_cs_get_count(np, con_id);
        if (ret > 0)
                return ret;
 
                        snprintf(propname, sizeof(propname), "%s",
                                 gpio_suffixes[i]);
 
-               ret = of_gpio_named_count(dev->of_node, propname);
+               ret = of_gpio_named_count(np, propname);
                if (ret > 0)
                        break;
        }
 
 #include <linux/notifier.h>
 
 struct device;
+struct fwnode_handle;
 
 struct gpio_chip;
 struct gpio_desc;
                               unsigned long *lookupflags);
 int of_gpiochip_add(struct gpio_chip *gc);
 void of_gpiochip_remove(struct gpio_chip *gc);
-int of_gpio_get_count(struct device *dev, const char *con_id);
+int of_gpio_count(const struct fwnode_handle *fwnode, const char *con_id);
 #else
 static inline struct gpio_desc *of_find_gpio(struct device_node *np,
                                             const char *con_id,
 }
 static inline int of_gpiochip_add(struct gpio_chip *gc) { return 0; }
 static inline void of_gpiochip_remove(struct gpio_chip *gc) { }
-static inline int of_gpio_get_count(struct device *dev, const char *con_id)
+static inline int of_gpio_count(const struct fwnode_handle *fwnode,
+                               const char *con_id)
 {
        return 0;
 }
 
        int count = -ENOENT;
 
        if (is_of_node(fwnode))
-               count = of_gpio_get_count(dev, con_id);
+               count = of_gpio_count(fwnode, con_id);
        else if (is_acpi_node(fwnode))
                count = acpi_gpio_count(dev, con_id);
        else if (is_software_node(fwnode))