#define NVME_SMALL_POOL_SIZE   256
 
 /*
- * These can be higher, but we need to ensure that any command doesn't
- * require an sg allocation that needs more than a page of data.
+ * Arbitrary upper bound.
  */
-#define NVME_MAX_KB_SZ 8192
+#define NVME_MAX_BYTES         SZ_8M
 #define NVME_MAX_NR_DESCRIPTORS        5
 
 /*
  */
 static __always_inline int nvme_pci_npages_prp(void)
 {
-       unsigned max_bytes = (NVME_MAX_KB_SZ * 1024) + NVME_CTRL_PAGE_SIZE;
+       unsigned max_bytes = NVME_MAX_BYTES + NVME_CTRL_PAGE_SIZE;
        unsigned nprps = DIV_ROUND_UP(max_bytes, NVME_CTRL_PAGE_SIZE);
        return DIV_ROUND_UP(8 * nprps, NVME_CTRL_PAGE_SIZE - 8);
 }
         * over a single page.
         */
        dev->ctrl.max_hw_sectors = min_t(u32,
-               NVME_MAX_KB_SZ << 1, dma_opt_mapping_size(&pdev->dev) >> 9);
+                       NVME_MAX_BYTES >> SECTOR_SHIFT,
+                       dma_opt_mapping_size(&pdev->dev) >> 9);
        dev->ctrl.max_segments = NVME_MAX_SEGS;
        dev->ctrl.max_integrity_segments = 1;
        return dev;