[PATCH net-next v4 7/9] gve: add netmem TX support to GVE DQO-RDA mode

[Mina Almasry]

Use netmem_dma_*() helpers in gve_tx_dqo.c DQO-RDA paths to
enable netmem TX support in that mode.

Declare support for netmem TX in GVE DQO-RDA mode.

Signed-off-by: Mina Almasry <almasrym...@google.com>

---

v4:
- New patch
---
 drivers/net/ethernet/google/gve/gve_main.c   | 4 ++++
 drivers/net/ethernet/google/gve/gve_tx_dqo.c | 8 +++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/drivers/net/ethernet/google/gve/gve_main.c 
b/drivers/net/ethernet/google/gve/gve_main.c
index 533e659b15b3..5047d4e260fb 100644
--- a/drivers/net/ethernet/google/gve/gve_main.c
+++ b/drivers/net/ethernet/google/gve/gve_main.c
@@ -2731,6 +2731,10 @@ static int gve_probe(struct pci_dev *pdev, const struct 
pci_device_id *ent)
 
        dev_info(&pdev->dev, "GVE version %s\n", gve_version_str);
        dev_info(&pdev->dev, "GVE queue format %d\n", (int)priv->queue_format);
+
+       if (!gve_is_gqi(priv) && !gve_is_qpl(priv))
+               dev->netmem_tx = true;
+
        gve_clear_probe_in_progress(priv);
        queue_work(priv->gve_wq, &priv->service_task);
        return 0;
diff --git a/drivers/net/ethernet/google/gve/gve_tx_dqo.c 
b/drivers/net/ethernet/google/gve/gve_tx_dqo.c
index 394debc62268..e74580dc7ebe 100644
--- a/drivers/net/ethernet/google/gve/gve_tx_dqo.c
+++ b/drivers/net/ethernet/google/gve/gve_tx_dqo.c
@@ -667,7 +667,8 @@ static int gve_tx_add_skb_no_copy_dqo(struct gve_tx_ring 
*tx,
                        goto err;
 
                dma_unmap_len_set(pkt, len[pkt->num_bufs], len);
-               dma_unmap_addr_set(pkt, dma[pkt->num_bufs], addr);
+               netmem_dma_unmap_addr_set(skb_frag_netmem(frag), pkt,
+                                         dma[pkt->num_bufs], addr);
                ++pkt->num_bufs;
 
                gve_tx_fill_pkt_desc_dqo(tx, desc_idx, skb, len, addr,
@@ -1045,8 +1046,9 @@ static void gve_unmap_packet(struct device *dev,
        dma_unmap_single(dev, dma_unmap_addr(pkt, dma[0]),
                         dma_unmap_len(pkt, len[0]), DMA_TO_DEVICE);
        for (i = 1; i < pkt->num_bufs; i++) {
-               dma_unmap_page(dev, dma_unmap_addr(pkt, dma[i]),
-                              dma_unmap_len(pkt, len[i]), DMA_TO_DEVICE);
+               netmem_dma_unmap_page_attrs(dev, dma_unmap_addr(pkt, dma[i]),
+                                           dma_unmap_len(pkt, len[i]),
+                                           DMA_TO_DEVICE, 0);
        }
        pkt->num_bufs = 0;
 }
-- 
2.48.1.601.g30ceb7b040-goog