habanalabs: Use simplified API for p2p dist calc

Use the simplified API that calculates distance between two devices.

Signed-off-by: Oded Gabbay <ogabbay@kernel.org>

diff --git a/drivers/misc/habanalabs/common/memory.c b/drivers/misc/habanalabs/common/memory.c
index ef28f3b37b93216bf9a7b50f296f21fcdff62261..99b1d6ce26aef4b0bff804f79c5132fbdff2fb55 100644 (file)
--- a/drivers/misc/habanalabs/common/memory.c
+++ b/drivers/misc/habanalabs/common/memory.c
@@ -1689,7 +1689,7 @@ static int hl_dmabuf_attach(struct dma_buf *dmabuf,
        hl_dmabuf = dmabuf->priv;
        hdev = hl_dmabuf->ctx->hdev;
 
-       rc = pci_p2pdma_distance_many(hdev->pdev, &attachment->dev, 1, true);
+       rc = pci_p2pdma_distance(hdev->pdev, attachment->dev, true);
 
        if (rc < 0)
                attachment->peer2peer = false;