synchronize_irq(adapter->pdev->irq);
 }
 
-#ifdef NETIF_F_HW_VLAN_TX
-static void atl2_vlan_rx_register(struct net_device *netdev,
-       struct vlan_group *grp)
+static void __atl2_vlan_mode(u32 features, u32 *ctrl)
+{
+       if (features & NETIF_F_HW_VLAN_RX) {
+               /* enable VLAN tag insert/strip */
+               *ctrl |= MAC_CTRL_RMV_VLAN;
+       } else {
+               /* disable VLAN tag insert/strip */
+               *ctrl &= ~MAC_CTRL_RMV_VLAN;
+       }
+}
+
+static void atl2_vlan_mode(struct net_device *netdev, u32 features)
 {
        struct atl2_adapter *adapter = netdev_priv(netdev);
        u32 ctrl;
 
        atl2_irq_disable(adapter);
-       adapter->vlgrp = grp;
 
-       if (grp) {
-               /* enable VLAN tag insert/strip */
-               ctrl = ATL2_READ_REG(&adapter->hw, REG_MAC_CTRL);
-               ctrl |= MAC_CTRL_RMV_VLAN;
-               ATL2_WRITE_REG(&adapter->hw, REG_MAC_CTRL, ctrl);
-       } else {
-               /* disable VLAN tag insert/strip */
-               ctrl = ATL2_READ_REG(&adapter->hw, REG_MAC_CTRL);
-               ctrl &= ~MAC_CTRL_RMV_VLAN;
-               ATL2_WRITE_REG(&adapter->hw, REG_MAC_CTRL, ctrl);
-       }
+       ctrl = ATL2_READ_REG(&adapter->hw, REG_MAC_CTRL);
+       __atl2_vlan_mode(features, &ctrl);
+       ATL2_WRITE_REG(&adapter->hw, REG_MAC_CTRL, ctrl);
 
        atl2_irq_enable(adapter);
 }
 
 static void atl2_restore_vlan(struct atl2_adapter *adapter)
 {
-       atl2_vlan_rx_register(adapter->netdev, adapter->vlgrp);
+       atl2_vlan_mode(adapter->netdev, adapter->netdev->features);
+}
+
+static u32 atl2_fix_features(struct net_device *netdev, u32 features)
+{
+       /*
+        * Since there is no support for separate rx/tx vlan accel
+        * enable/disable make sure tx flag is always in same state as rx.
+        */
+       if (features & NETIF_F_HW_VLAN_RX)
+               features |= NETIF_F_HW_VLAN_TX;
+       else
+               features &= ~NETIF_F_HW_VLAN_TX;
+
+       return features;
+}
+
+static int atl2_set_features(struct net_device *netdev, u32 features)
+{
+       u32 changed = netdev->features ^ features;
+
+       if (changed & NETIF_F_HW_VLAN_RX)
+               atl2_vlan_mode(netdev, features);
+
+       return 0;
 }
-#endif
 
 static void atl2_intr_rx(struct atl2_adapter *adapter)
 {
                        memcpy(skb->data, rxd->packet, rx_size);
                        skb_put(skb, rx_size);
                        skb->protocol = eth_type_trans(skb, netdev);
-#ifdef NETIF_F_HW_VLAN_TX
-                       if (adapter->vlgrp && (rxd->status.vlan)) {
+                       if (rxd->status.vlan) {
                                u16 vlan_tag = (rxd->status.vtag>>4) |
                                        ((rxd->status.vtag&7) << 13) |
                                        ((rxd->status.vtag&8) << 9);
-                               vlan_hwaccel_rx(skb, adapter->vlgrp, vlan_tag);
-                       } else
-#endif
+
+                               __vlan_hwaccel_put_tag(skb, vlan_tag);
+                       }
                        netif_rx(skb);
                        netdev->stats.rx_bytes += rx_size;
                        netdev->stats.rx_packets++;
        atl2_set_multi(netdev);
        init_ring_ptrs(adapter);
 
-#ifdef NETIF_F_HW_VLAN_TX
        atl2_restore_vlan(adapter);
-#endif
 
        if (atl2_configure(adapter)) {
                err = -EIO;
        atl2_set_multi(netdev);
        init_ring_ptrs(adapter);
 
-#ifdef NETIF_F_HW_VLAN_TX
        atl2_restore_vlan(adapter);
-#endif
 
        if (atl2_configure(adapter)) {
                err = -EIO;
                MAC_CTRL_PRMLEN_SHIFT);
 
        /* vlan */
-       if (adapter->vlgrp)
-               value |= MAC_CTRL_RMV_VLAN;
+       __atl2_vlan_mode(netdev->features, &value);
 
        /* filter mode */
        value |= MAC_CTRL_BC_EN;
        .ndo_validate_addr      = eth_validate_addr,
        .ndo_set_mac_address    = atl2_set_mac,
        .ndo_change_mtu         = atl2_change_mtu,
+       .ndo_fix_features       = atl2_fix_features,
+       .ndo_set_features       = atl2_set_features,
        .ndo_do_ioctl           = atl2_ioctl,
        .ndo_tx_timeout         = atl2_tx_timeout,
-       .ndo_vlan_rx_register   = atl2_vlan_rx_register,
 #ifdef CONFIG_NET_POLL_CONTROLLER
        .ndo_poll_controller    = atl2_poll_controller,
 #endif
 
        err = -EIO;
 
-       netdev->hw_features = NETIF_F_SG;
+       netdev->hw_features = NETIF_F_SG | NETIF_F_HW_VLAN_RX;
        netdev->features |= (NETIF_F_HW_VLAN_TX | NETIF_F_HW_VLAN_RX);
 
        /* Init PHY as early as possible due to power saving issue  */