unsigned long start_address;
        unsigned long start_pa;
        unsigned long last_pa;
+       unsigned long page_size;
        unsigned int level;
        u64 current_flags;
        bool check_wx;
 #endif
 
        pt_dump_seq_printf(st->seq, REG "-" REG " ", st->start_address, addr - 1);
-       if (st->start_pa == st->last_pa && st->start_address + PAGE_SIZE != addr) {
+       if (st->start_pa == st->last_pa && st->start_address + st->page_size != addr) {
                pt_dump_seq_printf(st->seq, "[" REG "]", st->start_pa);
-               delta = PAGE_SIZE >> 10;
+               delta = st->page_size >> 10;
        } else {
                pt_dump_seq_printf(st->seq, " " REG " ", st->start_pa);
                delta = (addr - st->start_address) >> 10;
 }
 
 static void note_page(struct pg_state *st, unsigned long addr,
-              unsigned int level, u64 val)
+              unsigned int level, u64 val, unsigned long page_size)
 {
        u64 flag = val & pg_level[level].mask;
        u64 pa = val & PTE_RPN_MASK;
                st->start_address = addr;
                st->start_pa = pa;
                st->last_pa = pa;
+               st->page_size = page_size;
                pt_dump_seq_printf(st->seq, "---[ %s ]---\n", st->marker->name);
        /*
         * Dump the section of virtual memory when:
         */
        } else if (flag != st->current_flags || level != st->level ||
                   addr >= st->marker[1].start_address ||
-                  (pa != st->last_pa + PAGE_SIZE &&
+                  (pa != st->last_pa + st->page_size &&
                    (pa != st->start_pa || st->start_pa != st->last_pa))) {
 
                /* Check the PTE flags */
                st->start_address = addr;
                st->start_pa = pa;
                st->last_pa = pa;
+               st->page_size = page_size;
                st->current_flags = flag;
                st->level = level;
        } else {
 
        for (i = 0; i < PTRS_PER_PTE; i++, pte++) {
                addr = start + i * PAGE_SIZE;
-               note_page(st, addr, 4, pte_val(*pte));
+               note_page(st, addr, 4, pte_val(*pte), PAGE_SIZE);
 
        }
 }
                        /* pmd exists */
                        walk_pte(st, pmd, addr);
                else
-                       note_page(st, addr, 3, pmd_val(*pmd));
+                       note_page(st, addr, 3, pmd_val(*pmd), PMD_SIZE);
        }
 }
 
                        /* pud exists */
                        walk_pmd(st, pud, addr);
                else
-                       note_page(st, addr, 2, pud_val(*pud));
+                       note_page(st, addr, 2, pud_val(*pud), PUD_SIZE);
        }
 }
 
                        /* pgd exists */
                        walk_pud(st, pgd, addr);
                else
-                       note_page(st, addr, 1, pgd_val(*pgd));
+                       note_page(st, addr, 1, pgd_val(*pgd), PGDIR_SIZE);
        }
 }
 
 
        /* Traverse kernel page tables */
        walk_pagetables(&st);
-       note_page(&st, 0, 0, 0);
+       note_page(&st, 0, 0, 0, 0);
        return 0;
 }