1 files changed, 8 insertions, 4 deletions

diff --git a/net/dsa/slave.c b/net/dsa/slave.c
index 9a88035517a6..75d58229a4bd 100644
--- a/net/dsa/slave.c
+++ b/net/dsa/slave.c
@@ -1035,12 +1035,16 @@ static int dsa_slave_setup_tc_block(struct net_device *dev,
 static int dsa_slave_setup_tc(struct net_device *dev, enum tc_setup_type type,
                               void *type_data)
 {
-        switch (type) {
-        case TC_SETUP_BLOCK:
+        struct dsa_port *dp = dsa_slave_to_port(dev);
+        struct dsa_switch *ds = dp->ds;
+
+        if (type == TC_SETUP_BLOCK)
                 return dsa_slave_setup_tc_block(dev, type_data);
-        default:
+
+        if (!ds->ops->port_setup_tc)
                 return -EOPNOTSUPP;
-        }
+
+        return ds->ops->port_setup_tc(ds, dp->index, type, type_data);
 }
 
 static void dsa_slave_get_stats64(struct net_device *dev,