u16 vid;
        u16 port_index;
        int row;
+       bool lag;
 };
 
 struct lan966x_mac_raw_entry {
        lan_wr(mach, lan966x, ANA_MACHDATA);
 }
 
-static int __lan966x_mac_learn(struct lan966x *lan966x, int pgid,
-                              bool cpu_copy,
-                              const unsigned char mac[ETH_ALEN],
-                              unsigned int vid,
-                              enum macaccess_entry_type type)
+static int __lan966x_mac_learn_locked(struct lan966x *lan966x, int pgid,
+                                     bool cpu_copy,
+                                     const unsigned char mac[ETH_ALEN],
+                                     unsigned int vid,
+                                     enum macaccess_entry_type type)
 {
-       int ret;
+       lockdep_assert_held(&lan966x->mac_lock);
 
-       spin_lock(&lan966x->mac_lock);
        lan966x_mac_select(lan966x, mac, vid);
 
        /* Issue a write command */
               ANA_MACACCESS_MAC_TABLE_CMD_SET(MACACCESS_CMD_LEARN),
               lan966x, ANA_MACACCESS);
 
-       ret = lan966x_mac_wait_for_completion(lan966x);
+       return lan966x_mac_wait_for_completion(lan966x);
+}
+
+static int __lan966x_mac_learn(struct lan966x *lan966x, int pgid,
+                              bool cpu_copy,
+                              const unsigned char mac[ETH_ALEN],
+                              unsigned int vid,
+                              enum macaccess_entry_type type)
+{
+       int ret;
+
+       spin_lock(&lan966x->mac_lock);
+       ret = __lan966x_mac_learn_locked(lan966x, pgid, cpu_copy, mac, vid, type);
        spin_unlock(&lan966x->mac_lock);
 
        return ret;
        return __lan966x_mac_learn(lan966x, port, false, mac, vid, type);
 }
 
+static int lan966x_mac_learn_locked(struct lan966x *lan966x, int port,
+                                   const unsigned char mac[ETH_ALEN],
+                                   unsigned int vid,
+                                   enum macaccess_entry_type type)
+{
+       WARN_ON(type != ENTRYTYPE_NORMAL && type != ENTRYTYPE_LOCKED);
+
+       return __lan966x_mac_learn_locked(lan966x, port, false, mac, vid, type);
+}
+
 static int lan966x_mac_forget_locked(struct lan966x *lan966x,
                                     const unsigned char mac[ETH_ALEN],
                                     unsigned int vid,
        INIT_LIST_HEAD(&lan966x->mac_entries);
 }
 
-static struct lan966x_mac_entry *lan966x_mac_alloc_entry(const unsigned char *mac,
-                                                        u16 vid, u16 port_index)
+static struct lan966x_mac_entry *lan966x_mac_alloc_entry(struct lan966x_port *port,
+                                                        const unsigned char *mac,
+                                                        u16 vid)
 {
        struct lan966x_mac_entry *mac_entry;
 
 
        memcpy(mac_entry->mac, mac, ETH_ALEN);
        mac_entry->vid = vid;
-       mac_entry->port_index = port_index;
+       mac_entry->port_index = port->chip_port;
        mac_entry->row = LAN966X_MAC_INVALID_ROW;
+       mac_entry->lag = port->bond ? true : false;
        return mac_entry;
 }
 
                goto mac_learn;
        }
 
-       mac_entry = lan966x_mac_alloc_entry(addr, vid, port->chip_port);
+       mac_entry = lan966x_mac_alloc_entry(port, addr, vid);
        if (!mac_entry) {
                spin_unlock(&lan966x->mac_lock);
                return -ENOMEM;
        list_add_tail(&mac_entry->list, &lan966x->mac_entries);
        spin_unlock(&lan966x->mac_lock);
 
-       lan966x_fdb_call_notifiers(SWITCHDEV_FDB_OFFLOADED, addr, vid, port->dev);
+       lan966x_fdb_call_notifiers(SWITCHDEV_FDB_OFFLOADED, addr, vid,
+                                  port->bond ?: port->dev);
 
 mac_learn:
        lan966x_mac_learn(lan966x, port->chip_port, addr, vid, ENTRYTYPE_LOCKED);
        return 0;
 }
 
+void lan966x_mac_lag_replace_port_entry(struct lan966x *lan966x,
+                                       struct lan966x_port *src,
+                                       struct lan966x_port *dst)
+{
+       struct lan966x_mac_entry *mac_entry;
+
+       spin_lock(&lan966x->mac_lock);
+       list_for_each_entry(mac_entry, &lan966x->mac_entries, list) {
+               if (mac_entry->port_index == src->chip_port &&
+                   mac_entry->lag) {
+                       lan966x_mac_forget_locked(lan966x, mac_entry->mac,
+                                                 mac_entry->vid,
+                                                 ENTRYTYPE_LOCKED);
+
+                       lan966x_mac_learn_locked(lan966x, dst->chip_port,
+                                                mac_entry->mac, mac_entry->vid,
+                                                ENTRYTYPE_LOCKED);
+                       mac_entry->port_index = dst->chip_port;
+               }
+       }
+       spin_unlock(&lan966x->mac_lock);
+}
+
+void lan966x_mac_lag_remove_port_entry(struct lan966x *lan966x,
+                                      struct lan966x_port *src)
+{
+       struct lan966x_mac_entry *mac_entry, *tmp;
+
+       spin_lock(&lan966x->mac_lock);
+       list_for_each_entry_safe(mac_entry, tmp, &lan966x->mac_entries,
+                                list) {
+               if (mac_entry->port_index == src->chip_port &&
+                   mac_entry->lag) {
+                       lan966x_mac_forget_locked(lan966x, mac_entry->mac,
+                                                 mac_entry->vid,
+                                                 ENTRYTYPE_LOCKED);
+
+                       list_del(&mac_entry->list);
+                       kfree(mac_entry);
+               }
+       }
+       spin_unlock(&lan966x->mac_lock);
+}
+
 void lan966x_mac_purge_entries(struct lan966x *lan966x)
 {
        struct lan966x_mac_entry *mac_entry, *tmp;
        struct lan966x_mac_entry *mac_entry, *tmp;
        unsigned char mac[ETH_ALEN] __aligned(2);
        struct list_head mac_deleted_entries;
+       struct lan966x_port *port;
        u32 dest_idx;
        u32 column;
        u16 vid;
                /* Notify the bridge that the entry doesn't exist
                 * anymore in the HW
                 */
+               port = lan966x->ports[mac_entry->port_index];
                lan966x_mac_notifiers(SWITCHDEV_FDB_DEL_TO_BRIDGE,
                                      mac_entry->mac, mac_entry->vid,
-                                     lan966x->ports[mac_entry->port_index]->dev);
+                                     port->bond ?: port->dev);
                list_del(&mac_entry->list);
                kfree(mac_entry);
        }
                        continue;
                }
 
-               mac_entry = lan966x_mac_alloc_entry(mac, vid, dest_idx);
+               port = lan966x->ports[dest_idx];
+               mac_entry = lan966x_mac_alloc_entry(port, mac, vid);
                if (!mac_entry) {
                        spin_unlock(&lan966x->mac_lock);
                        return;
                spin_unlock(&lan966x->mac_lock);
 
                lan966x_mac_notifiers(SWITCHDEV_FDB_ADD_TO_BRIDGE,
-                                     mac, vid, lan966x->ports[dest_idx]->dev);
+                                     mac, vid, port->bond ?: port->dev);
        }
 }