if (!dev->q || !dev->ops)
                return -EINVAL;
 
+       ret = nvm_init(dev);
+       if (ret)
+               return ret;
+
        dev->dma_pool = dev->ops->create_dma_pool(dev, "ppalist");
        if (!dev->dma_pool) {
                pr_err("nvm: could not create dma pool\n");
+               nvm_free(dev);
                return -ENOMEM;
        }
 
-       ret = nvm_init(dev);
-       if (ret)
-               goto err_init;
-
        /* register device with a supported media manager */
        down_write(&nvm_lock);
        list_add(&dev->devices, &nvm_devices);
        up_write(&nvm_lock);
 
        return 0;
-err_init:
-       dev->ops->destroy_dma_pool(dev->dma_pool);
-       return ret;
 }
 EXPORT_SYMBOL(nvm_register);
 
 
        if (ns->noiob)
                nvme_set_chunk_size(ns);
        nvme_update_disk_info(disk, ns, id);
-       if (ns->ndev)
-               nvme_nvm_update_nvm_info(ns);
 #ifdef CONFIG_NVME_MULTIPATH
        if (ns->head->disk) {
                nvme_update_disk_info(ns->head->disk, ns, id);
        nvme_setup_streams_ns(ctrl, ns);
        nvme_set_disk_name(disk_name, ns, ctrl, &flags);
 
-       if ((ctrl->quirks & NVME_QUIRK_LIGHTNVM) && id->vs[0] == 0x1) {
-               if (nvme_nvm_register(ns, disk_name, node)) {
-                       dev_warn(ctrl->device, "LightNVM init failure\n");
-                       goto out_unlink_ns;
-               }
-       }
-
        disk = alloc_disk_node(0, node);
        if (!disk)
                goto out_unlink_ns;
 
        __nvme_revalidate_disk(disk, id);
 
+       if ((ctrl->quirks & NVME_QUIRK_LIGHTNVM) && id->vs[0] == 0x1) {
+               if (nvme_nvm_register(ns, disk_name, node)) {
+                       dev_warn(ctrl->device, "LightNVM init failure\n");
+                       goto out_put_disk;
+               }
+       }
+
        down_write(&ctrl->namespaces_rwsem);
        list_add_tail(&ns->list, &ctrl->namespaces);
        up_write(&ctrl->namespaces_rwsem);
        kfree(id);
 
        return;
+ out_put_disk:
+       put_disk(ns->disk);
  out_unlink_ns:
        mutex_lock(&ctrl->subsys->lock);
        list_del_rcu(&ns->siblings);
 
        }
 }
 
-void nvme_nvm_update_nvm_info(struct nvme_ns *ns)
-{
-       struct nvm_dev *ndev = ns->ndev;
-       struct nvm_geo *geo = &ndev->geo;
-
-       if (geo->version == NVM_OCSSD_SPEC_12)
-               return;
-
-       geo->csecs = 1 << ns->lba_shift;
-       geo->sos = ns->ms;
-}
-
 int nvme_nvm_register(struct nvme_ns *ns, char *disk_name, int node)
 {
        struct request_queue *q = ns->queue;
        struct nvm_dev *dev;
+       struct nvm_geo *geo;
 
        _nvme_nvm_check_size();
 
        if (!dev)
                return -ENOMEM;
 
+       /* Note that csecs and sos will be overridden if it is a 1.2 drive. */
+       geo = &dev->geo;
+       geo->csecs = 1 << ns->lba_shift;
+       geo->sos = ns->ms;
+
        dev->q = q;
        memcpy(dev->name, disk_name, DISK_NAME_LEN);
        dev->ops = &nvme_nvm_dev_ops;
 
 #endif /* CONFIG_NVME_MULTIPATH */
 
 #ifdef CONFIG_NVM
-void nvme_nvm_update_nvm_info(struct nvme_ns *ns);
 int nvme_nvm_register(struct nvme_ns *ns, char *disk_name, int node);
 void nvme_nvm_unregister(struct nvme_ns *ns);
 extern const struct attribute_group nvme_nvm_attr_group;
 int nvme_nvm_ioctl(struct nvme_ns *ns, unsigned int cmd, unsigned long arg);
 #else
-static inline void nvme_nvm_update_nvm_info(struct nvme_ns *ns) {};
 static inline int nvme_nvm_register(struct nvme_ns *ns, char *disk_name,
                                    int node)
 {