1 files changed, 15 insertions, 5 deletions

diff --git a/drivers/dma/ioat/pci.c b/drivers/dma/ioat/pci.c
index 2c8d560e6334..1d051cd045db 100644
--- a/drivers/dma/ioat/pci.c
+++ b/drivers/dma/ioat/pci.c
@@ -123,6 +123,7 @@ module_param(ioat_dca_enabled, int, 0644);
 MODULE_PARM_DESC(ioat_dca_enabled, "control support of dca service (default: 1)");
 
 struct kmem_cache *ioat2_cache;
+struct kmem_cache *ioat3_sed_cache;
 
 #define DRV_NAME "ioatdma"
 
@@ -207,9 +208,6 @@ static void ioat_remove(struct pci_dev *pdev)
         if (!device)
                 return;
 
-        if (device->version >= IOAT_VER_3_0)
-                ioat3_dma_remove(device);
-
         dev_err(&pdev->dev, "Removing dma and dca services\n");
         if (device->dca) {
                 unregister_dca_provider(device->dca, &pdev->dev);
@@ -221,7 +219,7 @@ static void ioat_remove(struct pci_dev *pdev)
 
 static int __init ioat_init_module(void)
 {
-        int err;
+        int err = -ENOMEM;
 
         pr_info("%s: Intel(R) QuickData Technology Driver %s\n",
                 DRV_NAME, IOAT_DMA_VERSION);
@@ -231,9 +229,21 @@ static int __init ioat_init_module(void)
         if (!ioat2_cache)
                 return -ENOMEM;
 
+        ioat3_sed_cache = KMEM_CACHE(ioat_sed_ent, 0);
+        if (!ioat3_sed_cache)
+                goto err_ioat2_cache;
+
         err = pci_register_driver(&ioat_pci_driver);
         if (err)
-                kmem_cache_destroy(ioat2_cache);
+                goto err_ioat3_cache;
+
+        return 0;
+
+ err_ioat3_cache:
+        kmem_cache_destroy(ioat3_sed_cache);
+
+ err_ioat2_cache:
+        kmem_cache_destroy(ioat2_cache);
 
         return err;
 }