def METHOD_NAME(self, param_list: List[torch.nn.Parameter]):
    """
    the time step is increased. param list is used between current and the next
    time step.
    Args:
        param_list (List[torch.nn.Parameter]): a list of torch parameters.
    """
    for p in param_list:
        if p not in self._param_step_dict:
            self._param_step_dict[p] = [self._preop_step]
        else:
            self._param_step_dict[p].append(self._preop_step)
        self._param_runtime_order.append(p)
    self._step_param_dict[self._preop_step] = param_list
    self._preop_step += 1