1 files changed, 8 insertions, 2 deletions

diff --git a/drivers/net/dsa/bcm_sf2.c b/drivers/net/dsa/bcm_sf2.c
index 18a79579141f..7d62802a66d5 100644
--- a/drivers/net/dsa/bcm_sf2.c
+++ b/drivers/net/dsa/bcm_sf2.c
@@ -136,8 +136,15 @@ static int bcm_sf2_sw_get_sset_count(struct dsa_switch *ds)
 }
 
 static char *bcm_sf2_sw_probe(struct device *dsa_dev, struct device *host_dev,
-                              int sw_addr)
+                              int sw_addr, void **_priv)
 {
+        struct bcm_sf2_priv *priv;
+
+        priv = devm_kzalloc(dsa_dev, sizeof(*priv), GFP_KERNEL);
+        if (!priv)
+                return NULL;
+        *_priv = priv;
+
         return "Broadcom Starfighter 2";
 }
 
@@ -1363,7 +1370,6 @@ static int bcm_sf2_sw_set_wol(struct dsa_switch *ds, int port,
 
 static struct dsa_switch_driver bcm_sf2_switch_driver = {
         .tag_protocol           = DSA_TAG_PROTO_BRCM,
-        .priv_size              = sizeof(struct bcm_sf2_priv),
         .probe                  = bcm_sf2_sw_probe,
         .setup                  = bcm_sf2_sw_setup,
         .set_addr               = bcm_sf2_sw_set_addr,