choice
        prompt "Default allocator"
        depends on ZSWAP
-       default ZSWAP_ZPOOL_DEFAULT_ZSMALLOC if HAVE_ZSMALLOC
+       default ZSWAP_ZPOOL_DEFAULT_ZSMALLOC if MMU
        default ZSWAP_ZPOOL_DEFAULT_ZBUD
        help
          Selects the default allocator for the compressed cache for
 
 config ZSWAP_ZPOOL_DEFAULT_ZSMALLOC
        bool "zsmalloc"
-       depends on HAVE_ZSMALLOC
        select ZSMALLOC
        help
          Use the zsmalloc allocator as the default allocator.
          page. It is a ZBUD derivative so the simplicity and determinism are
          still there.
 
-config HAVE_ZSMALLOC
-       def_bool y
-       depends on MMU
-       depends on PAGE_SIZE_LESS_THAN_256KB # we want <= 64 KiB
-
 config ZSMALLOC
        tristate
        prompt "N:1 compression allocator (zsmalloc)" if ZSWAP
-       depends on HAVE_ZSMALLOC
+       depends on MMU
        help
          zsmalloc is a slab-based memory allocator designed to store
          pages of various compression levels efficiently. It achieves
 
  *     page->index: links together all component pages of a zspage
  *             For the huge page, this is always 0, so we use this field
  *             to store handle.
- *     page->page_type: PG_zsmalloc, lower 16 bit locate the first object
+ *     page->page_type: PGTY_zsmalloc, lower 24 bits locate the first object
  *             offset in a subpage of a zspage
  *
  * Usage of struct page flags:
        return first_page;
 }
 
-#define FIRST_OBJ_PAGE_TYPE_MASK       0xffff
-
-static inline void reset_first_obj_offset(struct page *page)
-{
-       VM_WARN_ON_ONCE(!PageZsmalloc(page));
-       page->page_type |= FIRST_OBJ_PAGE_TYPE_MASK;
-}
+#define FIRST_OBJ_PAGE_TYPE_MASK       0xffffff
 
 static inline unsigned int get_first_obj_offset(struct page *page)
 {
 
 static inline void set_first_obj_offset(struct page *page, unsigned int offset)
 {
-       /* With 16 bit available, we can support offsets into 64 KiB pages. */
-       BUILD_BUG_ON(PAGE_SIZE > SZ_64K);
+       /* With 24 bits available, we can support offsets into 16 MiB pages. */
+       BUILD_BUG_ON(PAGE_SIZE > SZ_16M);
        VM_WARN_ON_ONCE(!PageZsmalloc(page));
        VM_WARN_ON_ONCE(offset & ~FIRST_OBJ_PAGE_TYPE_MASK);
        page->page_type &= ~FIRST_OBJ_PAGE_TYPE_MASK;
        ClearPagePrivate(page);
        set_page_private(page, 0);
        page->index = 0;
-       reset_first_obj_offset(page);
        __ClearPageZsmalloc(page);
 }