static void nvme_set_chunk_size(struct nvme_ns *ns)
 {
-       u32 chunk_size = (((u32)ns->noiob) << (ns->lba_shift - 9));
+       u32 chunk_size = nvme_lba_to_sect(ns, ns->noiob);
        blk_queue_chunk_sectors(ns->queue, rounddown_pow_of_two(chunk_size));
 }
 
 
 static void nvme_config_write_zeroes(struct gendisk *disk, struct nvme_ns *ns)
 {
-       u32 max_sectors;
-       unsigned short bs = 1 << ns->lba_shift;
+       u64 max_blocks;
 
        if (!(ns->ctrl->oncs & NVME_CTRL_ONCS_WRITE_ZEROES) ||
            (ns->ctrl->quirks & NVME_QUIRK_DISABLE_WRITE_ZEROES))
         * nvme_init_identify() if available.
         */
        if (ns->ctrl->max_hw_sectors == UINT_MAX)
-               max_sectors = ((u32)(USHRT_MAX + 1) * bs) >> 9;
+               max_blocks = (u64)USHRT_MAX + 1;
        else
-               max_sectors = ((u32)(ns->ctrl->max_hw_sectors + 1) * bs) >> 9;
+               max_blocks = ns->ctrl->max_hw_sectors + 1;
 
-       blk_queue_max_write_zeroes_sectors(disk->queue, max_sectors);
+       blk_queue_max_write_zeroes_sectors(disk->queue,
+                                          nvme_lba_to_sect(ns, max_blocks));
 }
 
 static int nvme_report_ns_ids(struct nvme_ctrl *ctrl, unsigned int nsid,
 static void nvme_update_disk_info(struct gendisk *disk,
                struct nvme_ns *ns, struct nvme_id_ns *id)
 {
-       sector_t capacity = le64_to_cpu(id->nsze) << (ns->lba_shift - 9);
+       sector_t capacity = nvme_lba_to_sect(ns, le64_to_cpu(id->nsze));
        unsigned short bs = 1 << ns->lba_shift;
        u32 atomic_bs, phys_bs, io_opt;
 
 
        return sector >> (ns->lba_shift - SECTOR_SHIFT);
 }
 
+/*
+ * Convert a device logical block number to a 512B sector number.
+ */
+static inline sector_t nvme_lba_to_sect(struct nvme_ns *ns, u64 lba)
+{
+       return lba << (ns->lba_shift - SECTOR_SHIFT);
+}
+
 static inline void nvme_end_request(struct request *req, __le16 status,
                union nvme_result result)
 {