" Not in use with device managed"
                                         " flow steering");
 
-#define MLX4_VF                                        (1 << 0)
-
 #define HCA_GLOBAL_CAP_MASK            0
 #define PF_CONTEXT_BEHAVIOUR_MASK      0
 
        iounmap(owner);
 }
 
-static int __mlx4_init_one(struct pci_dev *pdev, const struct pci_device_id *id)
+static int __mlx4_init_one(struct pci_dev *pdev, int pci_dev_data)
 {
        struct mlx4_priv *priv;
        struct mlx4_dev *dev;
        /*
         * Check for BARs.
         */
-       if (((id == NULL) || !(id->driver_data & MLX4_VF)) &&
+       if (!(pci_dev_data & MLX4_PCI_DEV_IS_VF) &&
            !(pci_resource_flags(pdev, 0) & IORESOURCE_MEM)) {
                dev_err(&pdev->dev, "Missing DCS, aborting."
-                       "(id == 0X%p, id->driver_data: 0x%lx,"
-                       " pci_resource_flags(pdev, 0):0x%lx)\n", id,
-                       id ? id->driver_data : 0, pci_resource_flags(pdev, 0));
+                       "(driver_data: 0x%x, pci_resource_flags(pdev, 0):0x%lx)\n",
+                       pci_dev_data, pci_resource_flags(pdev, 0));
                err = -ENODEV;
                goto err_disable_pdev;
        }
 
        dev->rev_id = pdev->revision;
        /* Detect if this device is a virtual function */
-       if (id && id->driver_data & MLX4_VF) {
+       if (pci_dev_data & MLX4_PCI_DEV_IS_VF) {
                /* When acting as pf, we normally skip vfs unless explicitly
                 * requested to probe them. */
                if (num_vfs && extended_func_num(pdev) > probe_vf) {
        mlx4_sense_init(dev);
        mlx4_start_sense(dev);
 
+       priv->pci_dev_data = pci_dev_data;
        pci_set_drvdata(pdev, dev);
 
        return 0;
 {
        printk_once(KERN_INFO "%s", mlx4_version);
 
-       return __mlx4_init_one(pdev, id);
+       return __mlx4_init_one(pdev, id->driver_data);
 }
 
 static void mlx4_remove_one(struct pci_dev *pdev)
 
 int mlx4_restart_one(struct pci_dev *pdev)
 {
+       struct mlx4_dev  *dev  = pci_get_drvdata(pdev);
+       struct mlx4_priv *priv = mlx4_priv(dev);
+       int               pci_dev_data;
+
+       pci_dev_data = priv->pci_dev_data;
        mlx4_remove_one(pdev);
-       return __mlx4_init_one(pdev, NULL);
+       return __mlx4_init_one(pdev, pci_dev_data);
 }
 
 static DEFINE_PCI_DEVICE_TABLE(mlx4_pci_table) = {
        /* MT26478 ConnectX2 40GigE PCIe gen2 */
        { PCI_VDEVICE(MELLANOX, 0x676e), 0 },
        /* MT25400 Family [ConnectX-2 Virtual Function] */
-       { PCI_VDEVICE(MELLANOX, 0x1002), MLX4_VF },
+       { PCI_VDEVICE(MELLANOX, 0x1002), MLX4_PCI_DEV_IS_VF },
        /* MT27500 Family [ConnectX-3] */
        { PCI_VDEVICE(MELLANOX, 0x1003), 0 },
        /* MT27500 Family [ConnectX-3 Virtual Function] */
-       { PCI_VDEVICE(MELLANOX, 0x1004), MLX4_VF },
+       { PCI_VDEVICE(MELLANOX, 0x1004), MLX4_PCI_DEV_IS_VF },
        { PCI_VDEVICE(MELLANOX, 0x1005), 0 }, /* MT27510 Family */
        { PCI_VDEVICE(MELLANOX, 0x1006), 0 }, /* MT27511 Family */
        { PCI_VDEVICE(MELLANOX, 0x1007), 0 }, /* MT27520 Family */
 
 static pci_ers_result_t mlx4_pci_slot_reset(struct pci_dev *pdev)
 {
-       int ret = __mlx4_init_one(pdev, NULL);
+       int ret = __mlx4_init_one(pdev, 0);
 
        return ret ? PCI_ERS_RESULT_DISCONNECT : PCI_ERS_RESULT_RECOVERED;
 }