* adjust/expand SWIOTLB size for their use.
         */
        if (!io_tlb_nslabs) {
-               size = ALIGN(new_size, 1 << IO_TLB_SHIFT);
+               size = ALIGN(new_size, IO_TLB_SIZE);
                io_tlb_nslabs = size >> IO_TLB_SHIFT;
                io_tlb_nslabs = ALIGN(io_tlb_nslabs, IO_TLB_SEGSIZE);
 
 
        tbl_dma_addr &= mask;
 
-       offset_slots = ALIGN(tbl_dma_addr, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT;
+       offset_slots = ALIGN(tbl_dma_addr, IO_TLB_SIZE) >> IO_TLB_SHIFT;
 
        /*
         * Carefully handle integer overflow which can occur when mask == ~0UL.
         */
        max_slots = mask + 1
-                   ? ALIGN(mask + 1, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT
+                   ? ALIGN(mask + 1, IO_TLB_SIZE) >> IO_TLB_SHIFT
                    : 1UL << (BITS_PER_LONG - IO_TLB_SHIFT);
 
        /*
         * For mappings greater than or equal to a page, we limit the stride
         * (and hence alignment) to a page size.
         */
-       nslots = ALIGN(alloc_size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT;
+       nslots = ALIGN(alloc_size, IO_TLB_SIZE) >> IO_TLB_SHIFT;
        if (alloc_size >= PAGE_SIZE)
                stride = (1 << (PAGE_SHIFT - IO_TLB_SHIFT));
        else
                              enum dma_data_direction dir, unsigned long attrs)
 {
        unsigned long flags;
-       int i, count, nslots = ALIGN(alloc_size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT;
+       int i, count, nslots = ALIGN(alloc_size, IO_TLB_SIZE) >> IO_TLB_SHIFT;
        int index = (tlb_addr - io_tlb_start) >> IO_TLB_SHIFT;
        phys_addr_t orig_addr = io_tlb_orig_addr[index];
 
 
        if (orig_addr == INVALID_PHYS_ADDR)
                return;
-       orig_addr += (unsigned long)tlb_addr & ((1 << IO_TLB_SHIFT) - 1);
+       orig_addr += (unsigned long)tlb_addr & (IO_TLB_SIZE - 1);
 
        switch (target) {
        case SYNC_FOR_CPU:
 
 size_t swiotlb_max_mapping_size(struct device *dev)
 {
-       return ((size_t)1 << IO_TLB_SHIFT) * IO_TLB_SEGSIZE;
+       return ((size_t)IO_TLB_SIZE) * IO_TLB_SEGSIZE;
 }
 
 bool is_swiotlb_active(void)