1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/crypto/caam/jr.c b/drivers/crypto/caam/jr.c
index 6ce4c41c8634..9f16b2c03613 100644
--- a/drivers/crypto/caam/jr.c
+++ b/drivers/crypto/caam/jr.c
@@ -503,6 +503,14 @@ int caam_jr_probe(struct platform_device *pdev, struct device_node *np,
         dev_set_drvdata(jrdev, jrpriv);
         ctrlpriv->jrdev[ring] = jrdev;
 
+        if (sizeof(dma_addr_t) == sizeof(u64))
+                if (of_device_is_compatible(np, "fsl,sec-v5.0-job-ring"))
+                        dma_set_mask(jrdev, DMA_BIT_MASK(40));
+                else
+                        dma_set_mask(jrdev, DMA_BIT_MASK(36));
+        else
+                dma_set_mask(jrdev, DMA_BIT_MASK(32));
+
         /* Identify the interrupt */
         jrpriv->irq = of_irq_to_resource(np, 0, NULL);