return mlxsw_driver->trap_group_init(mlxsw_core, group);
 }
 
+static int
+mlxsw_devlink_trap_group_set(struct devlink *devlink,
+                            const struct devlink_trap_group *group,
+                            const struct devlink_trap_policer *policer)
+{
+       struct mlxsw_core *mlxsw_core = devlink_priv(devlink);
+       struct mlxsw_driver *mlxsw_driver = mlxsw_core->driver;
+
+       if (!mlxsw_driver->trap_group_set)
+               return -EOPNOTSUPP;
+       return mlxsw_driver->trap_group_set(mlxsw_core, group, policer);
+}
+
 static int
 mlxsw_devlink_trap_policer_init(struct devlink *devlink,
                                const struct devlink_trap_policer *policer)
        .trap_fini                      = mlxsw_devlink_trap_fini,
        .trap_action_set                = mlxsw_devlink_trap_action_set,
        .trap_group_init                = mlxsw_devlink_trap_group_init,
+       .trap_group_set                 = mlxsw_devlink_trap_group_set,
        .trap_policer_init              = mlxsw_devlink_trap_policer_init,
        .trap_policer_fini              = mlxsw_devlink_trap_policer_fini,
        .trap_policer_set               = mlxsw_devlink_trap_policer_set,
 
                               enum devlink_trap_action action);
        int (*trap_group_init)(struct mlxsw_core *mlxsw_core,
                               const struct devlink_trap_group *group);
+       int (*trap_group_set)(struct mlxsw_core *mlxsw_core,
+                             const struct devlink_trap_group *group,
+                             const struct devlink_trap_policer *policer);
        int (*trap_policer_init)(struct mlxsw_core *mlxsw_core,
                                 const struct devlink_trap_policer *policer);
        void (*trap_policer_fini)(struct mlxsw_core *mlxsw_core,
 
        .trap_fini                      = mlxsw_sp_trap_fini,
        .trap_action_set                = mlxsw_sp_trap_action_set,
        .trap_group_init                = mlxsw_sp_trap_group_init,
+       .trap_group_set                 = mlxsw_sp_trap_group_set,
        .trap_policer_init              = mlxsw_sp_trap_policer_init,
        .trap_policer_fini              = mlxsw_sp_trap_policer_fini,
        .trap_policer_set               = mlxsw_sp_trap_policer_set,
        .trap_fini                      = mlxsw_sp_trap_fini,
        .trap_action_set                = mlxsw_sp_trap_action_set,
        .trap_group_init                = mlxsw_sp_trap_group_init,
+       .trap_group_set                 = mlxsw_sp_trap_group_set,
        .trap_policer_init              = mlxsw_sp_trap_policer_init,
        .trap_policer_fini              = mlxsw_sp_trap_policer_fini,
        .trap_policer_set               = mlxsw_sp_trap_policer_set,
        .trap_fini                      = mlxsw_sp_trap_fini,
        .trap_action_set                = mlxsw_sp_trap_action_set,
        .trap_group_init                = mlxsw_sp_trap_group_init,
+       .trap_group_set                 = mlxsw_sp_trap_group_set,
        .trap_policer_init              = mlxsw_sp_trap_policer_init,
        .trap_policer_fini              = mlxsw_sp_trap_policer_fini,
        .trap_policer_set               = mlxsw_sp_trap_policer_set,
 
                             enum devlink_trap_action action);
 int mlxsw_sp_trap_group_init(struct mlxsw_core *mlxsw_core,
                             const struct devlink_trap_group *group);
+int mlxsw_sp_trap_group_set(struct mlxsw_core *mlxsw_core,
+                           const struct devlink_trap_group *group,
+                           const struct devlink_trap_policer *policer);
 int
 mlxsw_sp_trap_policer_init(struct mlxsw_core *mlxsw_core,
                           const struct devlink_trap_policer *policer);
 
        return 0;
 }
 
-int mlxsw_sp_trap_group_init(struct mlxsw_core *mlxsw_core,
-                            const struct devlink_trap_group *group)
+static int
+__mlxsw_sp_trap_group_init(struct mlxsw_core *mlxsw_core,
+                          const struct devlink_trap_group *group,
+                          u32 policer_id)
 {
        struct mlxsw_sp *mlxsw_sp = mlxsw_core_driver_priv(mlxsw_core);
        u16 hw_policer_id = MLXSW_REG_HTGT_INVALID_POLICER;
                return -EINVAL;
        }
 
-       if (group->init_policer_id) {
+       if (policer_id) {
                struct mlxsw_sp_trap_policer_item *policer_item;
-               u32 id = group->init_policer_id;
 
-               policer_item = mlxsw_sp_trap_policer_item_lookup(mlxsw_sp, id);
+               policer_item = mlxsw_sp_trap_policer_item_lookup(mlxsw_sp,
+                                                                policer_id);
                if (WARN_ON(!policer_item))
                        return -EINVAL;
                hw_policer_id = policer_item->hw_id;
        return mlxsw_reg_write(mlxsw_core, MLXSW_REG(htgt), htgt_pl);
 }
 
+int mlxsw_sp_trap_group_init(struct mlxsw_core *mlxsw_core,
+                            const struct devlink_trap_group *group)
+{
+       return __mlxsw_sp_trap_group_init(mlxsw_core, group,
+                                         group->init_policer_id);
+}
+
+int mlxsw_sp_trap_group_set(struct mlxsw_core *mlxsw_core,
+                           const struct devlink_trap_group *group,
+                           const struct devlink_trap_policer *policer)
+{
+       u32 policer_id = policer ? policer->id : 0;
+
+       return __mlxsw_sp_trap_group_init(mlxsw_core, group, policer_id);
+}
+
 static struct mlxsw_sp_trap_policer_item *
 mlxsw_sp_trap_policer_item_init(struct mlxsw_sp *mlxsw_sp, u32 id)
 {