status = 0;
        if (myri10ge_msi) {
                if (mgp->num_slices > 1) {
-                       status =
-                           pci_enable_msix(pdev, mgp->msix_vectors,
-                                           mgp->num_slices);
-                       if (status == 0) {
-                               mgp->msix_enabled = 1;
-                       } else {
+                       status = pci_enable_msix_range(pdev, mgp->msix_vectors,
+                                       mgp->num_slices, mgp->num_slices);
+                       if (status < 0) {
                                dev_err(&pdev->dev,
                                        "Error %d setting up MSI-X\n", status);
                                return status;
                        }
+                       mgp->msix_enabled = 1;
                }
                if (mgp->msix_enabled == 0) {
                        status = pci_enable_msi(pdev);
        mgp->msix_vectors = kcalloc(mgp->num_slices, sizeof(*mgp->msix_vectors),
                                    GFP_KERNEL);
        if (mgp->msix_vectors == NULL)
-               goto disable_msix;
+               goto no_msix;
        for (i = 0; i < mgp->num_slices; i++) {
                mgp->msix_vectors[i].entry = i;
        }
 
        while (mgp->num_slices > 1) {
-               /* make sure it is a power of two */
-               while (!is_power_of_2(mgp->num_slices))
-                       mgp->num_slices--;
+               mgp->num_slices = rounddown_pow_of_two(mgp->num_slices);
                if (mgp->num_slices == 1)
-                       goto disable_msix;
-               status = pci_enable_msix(pdev, mgp->msix_vectors,
-                                        mgp->num_slices);
-               if (status == 0) {
-                       pci_disable_msix(pdev);
+                       goto no_msix;
+               status = pci_enable_msix_range(pdev,
+                                              mgp->msix_vectors,
+                                              mgp->num_slices,
+                                              mgp->num_slices);
+               if (status < 0)
+                       goto no_msix;
+
+               pci_disable_msix(pdev);
+
+               if (status == mgp->num_slices) {
                        if (old_allocated)
                                kfree(old_fw);
                        return;
-               }
-               if (status > 0)
+               } else {
                        mgp->num_slices = status;
-               else
-                       goto disable_msix;
+               }
        }
 
-disable_msix:
+no_msix:
        if (mgp->msix_vectors != NULL) {
                kfree(mgp->msix_vectors);
                mgp->msix_vectors = NULL;