}
 }
 
-static void ocelot_setup_lag(struct ocelot *ocelot, int lag)
+/* When offloading a bonding interface, the switch ports configured under the
+ * same bond must have the same logical port ID, equal to the physical port ID
+ * of the lowest numbered physical port in that bond. Otherwise, in standalone/
+ * bridged mode, each port has a logical port ID equal to its physical port ID.
+ */
+static void ocelot_setup_logical_port_ids(struct ocelot *ocelot)
 {
-       unsigned long bond_mask = ocelot->lags[lag];
-       unsigned int p;
+       int port;
 
-       for_each_set_bit(p, &bond_mask, ocelot->num_phys_ports) {
-               u32 port_cfg = ocelot_read_gix(ocelot, ANA_PORT_PORT_CFG, p);
+       for (port = 0; port < ocelot->num_phys_ports; port++) {
+               struct ocelot_port *ocelot_port = ocelot->ports[port];
+               struct net_device *bond;
+
+               if (!ocelot_port)
+                       continue;
 
-               port_cfg &= ~ANA_PORT_PORT_CFG_PORTID_VAL_M;
+               bond = ocelot_port->bond;
+               if (bond) {
+                       int lag = __ffs(ocelot_get_bond_mask(ocelot, bond));
 
-               /* Use lag port as logical port for port i */
-               ocelot_write_gix(ocelot, port_cfg |
-                                ANA_PORT_PORT_CFG_PORTID_VAL(lag),
-                                ANA_PORT_PORT_CFG, p);
+                       ocelot_rmw_gix(ocelot,
+                                      ANA_PORT_PORT_CFG_PORTID_VAL(lag),
+                                      ANA_PORT_PORT_CFG_PORTID_VAL_M,
+                                      ANA_PORT_PORT_CFG, port);
+               } else {
+                       ocelot_rmw_gix(ocelot,
+                                      ANA_PORT_PORT_CFG_PORTID_VAL(port),
+                                      ANA_PORT_PORT_CFG_PORTID_VAL_M,
+                                      ANA_PORT_PORT_CFG, port);
+               }
        }
 }
 
                ocelot->lags[lag] |= BIT(port);
        }
 
-       ocelot_setup_lag(ocelot, lag);
+       ocelot_setup_logical_port_ids(ocelot);
        ocelot_apply_bridge_fwd_mask(ocelot);
        ocelot_set_aggr_pgids(ocelot);
 
 void ocelot_port_lag_leave(struct ocelot *ocelot, int port,
                           struct net_device *bond)
 {
-       u32 port_cfg;
        int i;
 
        ocelot->ports[port]->bond = NULL;
 
                ocelot->lags[n] = ocelot->lags[port];
                ocelot->lags[port] = 0;
-
-               ocelot_setup_lag(ocelot, n);
        }
 
-       port_cfg = ocelot_read_gix(ocelot, ANA_PORT_PORT_CFG, port);
-       port_cfg &= ~ANA_PORT_PORT_CFG_PORTID_VAL_M;
-       ocelot_write_gix(ocelot, port_cfg | ANA_PORT_PORT_CFG_PORTID_VAL(port),
-                        ANA_PORT_PORT_CFG, port);
-
+       ocelot_setup_logical_port_ids(ocelot);
        ocelot_apply_bridge_fwd_mask(ocelot);
        ocelot_set_aggr_pgids(ocelot);
 }