}
 }
 
-static void atl1e_vlan_rx_register(struct net_device *netdev,
-                                  struct vlan_group *grp)
+static void __atl1e_vlan_mode(u32 features, u32 *mac_ctrl_data)
+{
+       if (features & NETIF_F_HW_VLAN_RX) {
+               /* enable VLAN tag insert/strip */
+               *mac_ctrl_data |= MAC_CTRL_RMV_VLAN;
+       } else {
+               /* disable VLAN tag insert/strip */
+               *mac_ctrl_data &= ~MAC_CTRL_RMV_VLAN;
+       }
+}
+
+static void atl1e_vlan_mode(struct net_device *netdev, u32 features)
 {
        struct atl1e_adapter *adapter = netdev_priv(netdev);
        u32 mac_ctrl_data = 0;
        netdev_dbg(adapter->netdev, "%s\n", __func__);
 
        atl1e_irq_disable(adapter);
-
-       adapter->vlgrp = grp;
        mac_ctrl_data = AT_READ_REG(&adapter->hw, REG_MAC_CTRL);
-
-       if (grp) {
-               /* enable VLAN tag insert/strip */
-               mac_ctrl_data |= MAC_CTRL_RMV_VLAN;
-       } else {
-               /* disable VLAN tag insert/strip */
-               mac_ctrl_data &= ~MAC_CTRL_RMV_VLAN;
-       }
-
+       __atl1e_vlan_mode(features, &mac_ctrl_data);
        AT_WRITE_REG(&adapter->hw, REG_MAC_CTRL, mac_ctrl_data);
        atl1e_irq_enable(adapter);
 }
 static void atl1e_restore_vlan(struct atl1e_adapter *adapter)
 {
        netdev_dbg(adapter->netdev, "%s\n", __func__);
-       atl1e_vlan_rx_register(adapter->netdev, adapter->vlgrp);
+       atl1e_vlan_mode(adapter->netdev, adapter->netdev->features);
 }
+
 /*
  * atl1e_set_mac - Change the Ethernet Address of the NIC
  * @netdev: network interface device structure
        return 0;
 }
 
+static u32 atl1e_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 atl1e_set_features(struct net_device *netdev, u32 features)
+{
+       u32 changed = netdev->features ^ features;
+
+       if (changed & NETIF_F_HW_VLAN_RX)
+               atl1e_vlan_mode(netdev, features);
+
+       return 0;
+}
+
 /*
  * atl1e_change_mtu - Change the Maximum Transfer Unit
  * @netdev: network interface device structure
        value |= (((u32)adapter->hw.preamble_len &
                  MAC_CTRL_PRMLEN_MASK) << MAC_CTRL_PRMLEN_SHIFT);
 
-       if (adapter->vlgrp)
-               value |= MAC_CTRL_RMV_VLAN;
+       __atl1e_vlan_mode(netdev->features, &value);
 
        value |= MAC_CTRL_BC_EN;
        if (netdev->flags & IFF_PROMISC)
                        skb->protocol = eth_type_trans(skb, netdev);
                        atl1e_rx_checksum(adapter, skb, prrs);
 
-                       if (unlikely(adapter->vlgrp &&
-                               (prrs->pkt_flag & RRS_IS_VLAN_TAG))) {
+                       if (prrs->pkt_flag & RRS_IS_VLAN_TAG) {
                                u16 vlan_tag = (prrs->vtag >> 4) |
                                               ((prrs->vtag & 7) << 13) |
                                               ((prrs->vtag & 8) << 9);
                                netdev_dbg(netdev,
                                           "RXD VLAN TAG<RRD>=0x%04x\n",
                                           prrs->vtag);
-                               vlan_hwaccel_receive_skb(skb, adapter->vlgrp,
-                                                        vlan_tag);
-                       } else {
-                               netif_receive_skb(skb);
+                               __vlan_hwaccel_put_tag(skb, vlan_tag);
                        }
+                       netif_receive_skb(skb);
 
 skip_pkt:
        /* skip current packet whether it's ok or not. */
 
        tpd = atl1e_get_tpd(adapter);
 
-       if (unlikely(vlan_tx_tag_present(skb))) {
+       if (vlan_tx_tag_present(skb)) {
                u16 vlan_tag = vlan_tx_tag_get(skb);
                u16 atl1e_vlan_tag;
 
                                 MAC_CTRL_PRMLEN_MASK) <<
                                 MAC_CTRL_PRMLEN_SHIFT);
 
-               if (adapter->vlgrp)
-                       mac_ctrl_data |= MAC_CTRL_RMV_VLAN;
+               __atl1e_vlan_mode(netdev->features, &mac_ctrl_data);
 
                /* magic packet maybe Broadcast&multicast&Unicast frame */
                if (wufc & AT_WUFC_MAG)
        .ndo_set_multicast_list = atl1e_set_multi,
        .ndo_validate_addr      = eth_validate_addr,
        .ndo_set_mac_address    = atl1e_set_mac_addr,
+       .ndo_fix_features       = atl1e_fix_features,
+       .ndo_set_features       = atl1e_set_features,
        .ndo_change_mtu         = atl1e_change_mtu,
        .ndo_do_ioctl           = atl1e_ioctl,
        .ndo_tx_timeout         = atl1e_tx_timeout,
-       .ndo_vlan_rx_register   = atl1e_vlan_rx_register,
 #ifdef CONFIG_NET_POLL_CONTROLLER
        .ndo_poll_controller    = atl1e_netpoll,
 #endif
        atl1e_set_ethtool_ops(netdev);
 
        netdev->hw_features = NETIF_F_SG | NETIF_F_HW_CSUM | NETIF_F_TSO |
-               NETIF_F_HW_VLAN_TX;
-       netdev->features = netdev->hw_features |
-               NETIF_F_HW_VLAN_RX | NETIF_F_LLTX;
+                             NETIF_F_HW_VLAN_RX;
+       netdev->features = netdev->hw_features | NETIF_F_LLTX |
+                          NETIF_F_HW_VLAN_TX;
 
        return 0;
 }