#endif /* CONFIG_PCI_MSI */
 
-static int num_vfs;
-module_param(num_vfs, int, 0444);
-MODULE_PARM_DESC(num_vfs, "enable #num_vfs functions if num_vfs > 0");
-
-static int probe_vf;
-module_param(probe_vf, int, 0644);
-MODULE_PARM_DESC(probe_vf, "number of vfs to probe by pf driver (num_vfs > 0)");
+static uint8_t num_vfs[3] = {0, 0, 0};
+static int num_vfs_argc = 3;
+module_param_array(num_vfs, byte , &num_vfs_argc, 0444);
+MODULE_PARM_DESC(num_vfs, "enable #num_vfs functions if num_vfs > 0\n"
+                         "num_vfs=port1,port2,port1+2");
+
+static uint8_t probe_vf[3] = {0, 0, 0};
+static int probe_vfs_argc = 3;
+module_param_array(probe_vf, byte, &probe_vfs_argc, 0444);
+MODULE_PARM_DESC(probe_vf, "number of vfs to probe by pf driver (num_vfs > 0)\n"
+                          "probe_vf=port1,port2,port1+2");
 
 int mlx4_log_num_mgm_entry_size = MLX4_DEFAULT_MGM_LOG_ENTRY_SIZE;
 module_param_named(log_num_mgm_entry_size,
        struct mlx4_dev *dev;
        int err;
        int port;
-       int nvfs[MLX4_MAX_PORTS + 1], prb_vf[MLX4_MAX_PORTS + 1];
+       int nvfs[MLX4_MAX_PORTS + 1] = {0, 0, 0};
+       int prb_vf[MLX4_MAX_PORTS + 1] = {0, 0, 0};
+       const int param_map[MLX4_MAX_PORTS + 1][MLX4_MAX_PORTS + 1] = {
+               {2, 0, 0}, {0, 1, 2}, {0, 1, 2} };
        unsigned total_vfs = 0;
        int sriov_initialized = 0;
        unsigned int i;
         * per port, we must limit the number of VFs to 63 (since their are
         * 128 MACs)
         */
-       for (i = 0; i < sizeof(nvfs)/sizeof(nvfs[0]);
-            total_vfs += nvfs[i], i++) {
-               nvfs[i] = i == MLX4_MAX_PORTS ? num_vfs : 0;
+       for (i = 0; i < sizeof(nvfs)/sizeof(nvfs[0]) && i < num_vfs_argc;
+            total_vfs += nvfs[param_map[num_vfs_argc - 1][i]], i++) {
+               nvfs[param_map[num_vfs_argc - 1][i]] = num_vfs[i];
                if (nvfs[i] < 0) {
                        dev_err(&pdev->dev, "num_vfs module parameter cannot be negative\n");
                        return -EINVAL;
                }
        }
-       for (i = 0; i < sizeof(prb_vf)/sizeof(prb_vf[0]); i++) {
-               prb_vf[i] = i == MLX4_MAX_PORTS ? probe_vf : 0;
+       for (i = 0; i < sizeof(prb_vf)/sizeof(prb_vf[0]) && i < probe_vfs_argc;
+            i++) {
+               prb_vf[param_map[probe_vfs_argc - 1][i]] = probe_vf[i];
                if (prb_vf[i] < 0 || prb_vf[i] > nvfs[i]) {
                        dev_err(&pdev->dev, "probe_vf module parameter cannot be negative or greater than num_vfs\n");
                        return -EINVAL;
                        goto err_close;
                }
                if (sriov_initialized) {
+                       int ib_ports = 0;
+                       mlx4_foreach_port(i, dev, MLX4_PORT_TYPE_IB)
+                               ib_ports++;
+
+                       if (ib_ports &&
+                           (num_vfs_argc > 1 || probe_vfs_argc > 1)) {
+                               mlx4_err(dev,
+                                        "Invalid syntax of num_vfs/probe_vfs "
+                                        "with IB port. Single port VFs syntax"
+                                        " is only supported when all ports "
+                                        "are configured as ethernet\n");
+                               goto err_close;
+                       }
                        for (i = 0; i < sizeof(nvfs)/sizeof(nvfs[0]); i++) {
                                unsigned j;
                                for (j = 0; j < nvfs[i]; ++sum, ++j) {