struct sg_table *sgt)
 {
        struct scatterlist *sg, *sg_next_iter;
-       u32 count, len, dma_desc_cnt, len_next;
+       u32 count, dma_desc_cnt;
+       u64 len, len_next;
        dma_addr_t addr, addr_next;
 
        dma_desc_cnt = 0;
 {
        struct hl_userptr *userptr;
        struct scatterlist *sg, *sg_next_iter;
-       u32 count, len, dma_desc_cnt, len_next;
+       u32 count, dma_desc_cnt;
+       u64 len, len_next;
        dma_addr_t dma_addr, dma_addr_next;
        enum goya_dma_direction user_dir;
        u64 device_memory_addr, addr;
 
 #define VA_DDR_SPACE_SIZE      (VA_DDR_SPACE_END - \
                                        VA_DDR_SPACE_START)     /* 128GB */
 
-#define DMA_MAX_TRANSFER_SIZE  0xFFFFFFFF
+#define DMA_MAX_TRANSFER_SIZE  U32_MAX
 
 #define HW_CAP_PLL             0x00000001
 #define HW_CAP_DDR_0           0x00000002