return flags;
 }
 
-static inline __sum16 gre_checksum(struct sk_buff *skb)
-{
-       __wsum csum;
-
-       if (skb->ip_summed == CHECKSUM_PARTIAL)
-               csum = lco_csum(skb);
-       else
-               csum = skb_checksum(skb, 0, skb->len, 0);
-       return csum_fold(csum);
-}
-
 static inline void gre_build_header(struct sk_buff *skb, int hdr_len,
                                    __be16 flags, __be16 proto,
                                    __be32 key, __be32 seq)
                    !(skb_shinfo(skb)->gso_type &
                      (SKB_GSO_GRE | SKB_GSO_GRE_CSUM))) {
                        *ptr = 0;
-                       *(__sum16 *)ptr = gre_checksum(skb);
+                       if (skb->ip_summed == CHECKSUM_PARTIAL) {
+                               *(__sum16 *)ptr = csum_fold(lco_csum(skb));
+                       } else {
+                               skb->ip_summed = CHECKSUM_PARTIAL;
+                               skb->csum_start = skb_transport_header(skb) - skb->head;
+                               skb->csum_offset = sizeof(*greh);
+                       }
                }
        }
 }
 
                                       netdev_features_t features)
 {
        int tnl_hlen = skb_inner_mac_header(skb) - skb_transport_header(skb);
+       bool need_csum, offload_csum, gso_partial, need_ipsec;
        struct sk_buff *segs = ERR_PTR(-EINVAL);
        u16 mac_offset = skb->mac_header;
        __be16 protocol = skb->protocol;
-       bool need_csum, gso_partial;
        u16 mac_len = skb->mac_len;
        int gre_offset, outer_hlen;
 
        if (need_csum)
                features &= ~NETIF_F_SCTP_CRC;
 
+       need_ipsec = skb_dst(skb) && dst_xfrm(skb_dst(skb));
+       /* Try to offload checksum if possible */
+       offload_csum = !!(need_csum && !need_ipsec &&
+                         (skb->dev->features & NETIF_F_HW_CSUM));
+
        /* segment inner packet. */
        segs = skb_mac_gso_segment(skb, features);
        if (IS_ERR_OR_NULL(segs)) {
                }
 
                *(pcsum + 1) = 0;
-               *pcsum = gso_make_checksum(skb, 0);
+               if (skb->encapsulation || !offload_csum) {
+                       *pcsum = gso_make_checksum(skb, 0);
+               } else {
+                       skb->ip_summed = CHECKSUM_PARTIAL;
+                       skb->csum_start = skb_transport_header(skb) - skb->head;
+                       skb->csum_offset = sizeof(*greh);
+               }
        } while ((skb = skb->next));
 out:
        return segs;