NVMe: Use dma_set_mask() correctly