2 files changed, 8 insertions, 2 deletions

diff --git a/drivers/net/ethernet/mellanox/mlxsw/spectrum.c b/drivers/net/ethernet/mellanox/mlxsw/spectrum.c
index 2d46ec84ebdf..2d0897b7d860 100644
--- a/drivers/net/ethernet/mellanox/mlxsw/spectrum.c
+++ b/drivers/net/ethernet/mellanox/mlxsw/spectrum.c
@@ -3142,13 +3142,17 @@ static int mlxsw_sp_ports_create(struct mlxsw_sp *mlxsw_sp)
         if (!mlxsw_sp->ports)
                 return -ENOMEM;
 
-        mlxsw_sp->port_to_module = kcalloc(max_ports, sizeof(u8), GFP_KERNEL);
+        mlxsw_sp->port_to_module = kmalloc_array(max_ports, sizeof(int),
+                                                 GFP_KERNEL);
         if (!mlxsw_sp->port_to_module) {
                 err = -ENOMEM;
                 goto err_port_to_module_alloc;
         }
 
         for (i = 1; i < max_ports; i++) {
+                /* Mark as invalid */
+                mlxsw_sp->port_to_module[i] = -1;
+
                 err = mlxsw_sp_port_module_info_get(mlxsw_sp, i, &module,
                                                     &width, &lane);
                 if (err)
@@ -3216,6 +3220,8 @@ static void mlxsw_sp_port_unsplit_create(struct mlxsw_sp *mlxsw_sp,
 
         for (i = 0; i < count; i++) {
                 local_port = base_port + i * 2;
+                if (mlxsw_sp->port_to_module[local_port] < 0)
+                        continue;
                 module = mlxsw_sp->port_to_module[local_port];
 
                 mlxsw_sp_port_create(mlxsw_sp, local_port, false, module,
diff --git a/drivers/net/ethernet/mellanox/mlxsw/spectrum.h b/drivers/net/ethernet/mellanox/mlxsw/spectrum.h
index 58cf222fb985..432ab9b12b7f 100644
--- a/drivers/net/ethernet/mellanox/mlxsw/spectrum.h
+++ b/drivers/net/ethernet/mellanox/mlxsw/spectrum.h
@@ -152,7 +152,7 @@ struct mlxsw_sp {
         const struct mlxsw_bus_info *bus_info;
         unsigned char base_mac[ETH_ALEN];
         struct mlxsw_sp_upper *lags;
-        u8 *port_to_module;
+        int *port_to_module;
         struct mlxsw_sp_sb *sb;
         struct mlxsw_sp_bridge *bridge;
         struct mlxsw_sp_router *router;