static int mlxsw_m_ports_create(struct mlxsw_m *mlxsw_m)
 {
        unsigned int max_ports = mlxsw_core_max_ports(mlxsw_m->core);
+       struct devlink *devlink = priv_to_devlink(mlxsw_m->core);
        u8 last_module = max_ports;
        int i;
        int err;
        }
 
        /* Create port objects for each valid entry */
+       devl_lock(devlink);
        for (i = 0; i < mlxsw_m->max_ports; i++) {
                if (mlxsw_m->module_to_port[i] > 0 &&
                    !mlxsw_core_port_is_xm(mlxsw_m->core, i)) {
                                goto err_module_to_port_create;
                }
        }
+       devl_unlock(devlink);
 
        return 0;
 
                        mlxsw_m_port_remove(mlxsw_m,
                                            mlxsw_m->module_to_port[i]);
        }
+       devl_unlock(devlink);
        i = max_ports;
 err_module_to_port_map:
        for (i--; i > 0; i--)
 
 static void mlxsw_m_ports_remove(struct mlxsw_m *mlxsw_m)
 {
+       struct devlink *devlink = priv_to_devlink(mlxsw_m->core);
        int i;
 
+       devl_lock(devlink);
        for (i = 0; i < mlxsw_m->max_ports; i++) {
                if (mlxsw_m->module_to_port[i] > 0) {
                        mlxsw_m_port_remove(mlxsw_m,
                        mlxsw_m_port_module_unmap(mlxsw_m, i);
                }
        }
+       devl_unlock(devlink);
 
        kfree(mlxsw_m->module_to_port);
        kfree(mlxsw_m->ports);
                        struct netlink_ext_ack *extack)
 {
        struct mlxsw_m *mlxsw_m = mlxsw_core_driver_priv(mlxsw_core);
-       struct devlink *devlink = priv_to_devlink(mlxsw_core);
        int err;
 
        mlxsw_m->core = mlxsw_core;
                return err;
        }
 
-       devl_lock(devlink);
        err = mlxsw_m_ports_create(mlxsw_m);
-       devl_unlock(devlink);
        if (err) {
                dev_err(mlxsw_m->bus_info->dev, "Failed to create ports\n");
                return err;
 static void mlxsw_m_fini(struct mlxsw_core *mlxsw_core)
 {
        struct mlxsw_m *mlxsw_m = mlxsw_core_driver_priv(mlxsw_core);
-       struct devlink *devlink = priv_to_devlink(mlxsw_core);
 
-       devl_lock(devlink);
        mlxsw_m_ports_remove(mlxsw_m);
-       devl_unlock(devlink);
 }
 
 static const struct mlxsw_config_profile mlxsw_m_config_profile;
 
 
 static void mlxsw_sp_ports_remove(struct mlxsw_sp *mlxsw_sp)
 {
+       struct devlink *devlink = priv_to_devlink(mlxsw_sp->core);
        int i;
 
+       devl_lock(devlink);
        for (i = 1; i < mlxsw_core_max_ports(mlxsw_sp->core); i++)
                if (mlxsw_sp_port_created(mlxsw_sp, i))
                        mlxsw_sp_port_remove(mlxsw_sp, i);
        mlxsw_sp_cpu_port_remove(mlxsw_sp);
+       devl_unlock(devlink);
        kfree(mlxsw_sp->ports);
        mlxsw_sp->ports = NULL;
 }
 static int mlxsw_sp_ports_create(struct mlxsw_sp *mlxsw_sp)
 {
        unsigned int max_ports = mlxsw_core_max_ports(mlxsw_sp->core);
+       struct devlink *devlink = priv_to_devlink(mlxsw_sp->core);
        struct mlxsw_sp_port_mapping *port_mapping;
        size_t alloc_size;
        int i;
        if (!mlxsw_sp->ports)
                return -ENOMEM;
 
+       devl_lock(devlink);
        err = mlxsw_sp_cpu_port_create(mlxsw_sp);
        if (err)
                goto err_cpu_port_create;
                if (err)
                        goto err_port_create;
        }
+       devl_unlock(devlink);
        return 0;
 
 err_port_create:
                        mlxsw_sp_port_remove(mlxsw_sp, i);
        mlxsw_sp_cpu_port_remove(mlxsw_sp);
 err_cpu_port_create:
+       devl_unlock(devlink);
        kfree(mlxsw_sp->ports);
        mlxsw_sp->ports = NULL;
        return err;
                         struct netlink_ext_ack *extack)
 {
        struct mlxsw_sp *mlxsw_sp = mlxsw_core_driver_priv(mlxsw_core);
-       struct devlink *devlink = priv_to_devlink(mlxsw_core);
        int err;
 
        mlxsw_sp->core = mlxsw_core;
                goto err_sample_trigger_init;
        }
 
-       devl_lock(devlink);
        err = mlxsw_sp_ports_create(mlxsw_sp);
-       devl_unlock(devlink);
        if (err) {
                dev_err(mlxsw_sp->bus_info->dev, "Failed to create ports\n");
                goto err_ports_create;
 static void mlxsw_sp_fini(struct mlxsw_core *mlxsw_core)
 {
        struct mlxsw_sp *mlxsw_sp = mlxsw_core_driver_priv(mlxsw_core);
-       struct devlink *devlink = priv_to_devlink(mlxsw_core);
 
-       devl_lock(devlink);
        mlxsw_sp_ports_remove(mlxsw_sp);
-       devl_unlock(devlink);
-
        rhashtable_destroy(&mlxsw_sp->sample_trigger_ht);
        mlxsw_sp_port_module_info_fini(mlxsw_sp);
        mlxsw_sp_dpipe_fini(mlxsw_sp);