habanalabs: set dma max segment size

This is required from any device that is capable to perform DMA.

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

diff --git a/drivers/misc/habanalabs/common/pci/pci.c b/drivers/misc/habanalabs/common/pci/pci.c
index d5bedf5ba011be7a0c294618daf2ae78f46662c8..0b5366cc84fdb39836c51f299ce2030b73e1af66 100644 (file)
--- a/drivers/misc/habanalabs/common/pci/pci.c
+++ b/drivers/misc/habanalabs/common/pci/pci.c
@@ -436,6 +436,8 @@ int hl_pci_init(struct hl_device *hdev)
                goto unmap_pci_bars;
        }
 
+       dma_set_max_seg_size(&pdev->dev, U32_MAX);
+
        return 0;
 
 unmap_pci_bars: