1 files changed, 14 insertions, 3 deletions

diff --git a/drivers/net/dsa/mv88e6352.c b/drivers/net/dsa/mv88e6352.c
index 577ab2cfa944..7081a78a67e1 100644
--- a/drivers/net/dsa/mv88e6352.c
+++ b/drivers/net/dsa/mv88e6352.c
@@ -38,10 +38,20 @@ static const struct mv88e6xxx_switch_id mv88e6352_table[] = {
 };
 
 static char *mv88e6352_probe(struct device *dsa_dev, struct device *host_dev,
-                             int sw_addr)
+                             int sw_addr, void **priv)
 {
-        return mv88e6xxx_lookup_name(host_dev, sw_addr, mv88e6352_table,
+        struct mv88e6xxx_priv_state *ps;
+        char *name;
+
+        name = mv88e6xxx_lookup_name(host_dev, sw_addr, mv88e6352_table,
                                      ARRAY_SIZE(mv88e6352_table));
+        if (name) {
+                ps = devm_kzalloc(dsa_dev, sizeof(*ps), GFP_KERNEL);
+                if (!ps)
+                        return NULL;
+                *priv = ps;
+        }
+        return name;
 }
 
 static int mv88e6352_setup_global(struct dsa_switch *ds)
@@ -82,6 +92,8 @@ static int mv88e6352_setup(struct dsa_switch *ds)
         struct mv88e6xxx_priv_state *ps = ds_to_priv(ds);
         int ret;
 
+        ps->ds = ds;
+
         ret = mv88e6xxx_setup_common(ds);
         if (ret < 0)
                 return ret;
@@ -303,7 +315,6 @@ static int mv88e6352_set_eeprom(struct dsa_switch *ds,
 
 struct dsa_switch_driver mv88e6352_switch_driver = {
         .tag_protocol           = DSA_TAG_PROTO_EDSA,
-        .priv_size              = sizeof(struct mv88e6xxx_priv_state),
         .probe                  = mv88e6352_probe,
         .setup                  = mv88e6352_setup,
         .set_addr               = mv88e6xxx_set_addr_indirect,