habanalabs: fix integer handling issue

Need to add ull suffix to constant when doing shift of constant
into 64-bit variables

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

diff --git a/drivers/misc/habanalabs/common/mmu/mmu.c b/drivers/misc/habanalabs/common/mmu/mmu.c
index 27dc0d1..71703a3 100644 (file)
--- a/drivers/misc/habanalabs/common/mmu/mmu.c
+++ b/drivers/misc/habanalabs/common/mmu/mmu.c
@@ -507,7 +507,7 @@ static void hl_mmu_pa_page_with_offset(struct hl_ctx *ctx, u64 virt_addr,
        p = (char *)p + hop0_shift_off;
        p = (char *)p + ((hops->used_hops - 1) * sizeof(u64));
        hop_shift = *(u64 *)p;
-       offset_mask = (1 << hop_shift) - 1;
+       offset_mask = (1ull << hop_shift) - 1;
        addr_mask = ~(offset_mask);
        *phys_addr = (tmp_phys_addr & addr_mask) |
                                        (virt_addr & offset_mask);