prune unused sequential epoch start event

Signed-off-by: Brian Dellabetta <[email protected]>

Author: brian-dellabetta

src/llmcompressor/core/events/event.py

@@ -53,7 +53,6 @@ class EventType(Enum):
 
     # calibration lifecycle
     CALIBRATION_EPOCH_START = "calibration_epoch_start"
-    SEQUENTIAL_EPOCH_START = "sequential_epoch_start"
     SEQUENTIAL_EPOCH_END = "sequential_epoch_end"
     CALIBRATION_EPOCH_END = "calibration_epoch_end"
 

src/llmcompressor/core/session_functions.py

@@ -146,17 +146,6 @@ def calibration_epoch_start(cls, **kwargs) -> ModifiedState:
         """
         return cls.event(EventType.CALIBRATION_EPOCH_START, **kwargs)
 
-    @classmethod
-    def sequential_epoch_start(cls, **kwargs) -> ModifiedState:
-        """
-        Invoke a sequential epoch start event for the active session. This event should
-        be called after one sequential layer has been calibrated/trained for one epoch
-
-        This is called after a sequential layer has been calibrated with one batch, see
-        `src/llmcompressor/pipelines/sequential/pipeline.py` for usage example
-        """
-        return cls.event(EventType.SEQUENTIAL_EPOCH_START, **kwargs)
-
     @classmethod
     def sequential_epoch_end(cls, **kwargs) -> ModifiedState:
         """

src/llmcompressor/modifiers/awq/base.py

@@ -217,9 +217,6 @@ def on_event(self, state: State, event: Event, **kwargs):
             if not self.started_:
                 self.on_start(state, None)
 
-        elif event.type_ == EventType.SEQUENTIAL_EPOCH_START:
-            pass
-
         elif event.type_ == EventType.SEQUENTIAL_EPOCH_END:
             # Run smoothing in case of sequential pipeline
             self._apply_smoothing(state.model)

src/llmcompressor/pipelines/layer_sequential/pipeline.py

@@ -73,8 +73,6 @@ def __call__(
                 calib_desc = f"({layer_index + 1}/{num_layers}): Calibrating"
                 prop_desc = f"({layer_index + 1}/{num_layers}): Propagating"
 
-                LifecycleCallbacks.sequential_epoch_start()
-
                 # do a preliminary pass to trigger modifier hooks
                 for batch_idx in tqdm.tqdm(range(len(dataloader)), desc=calib_desc):
                     inputs = intermediates.fetch(batch_idx)

src/llmcompressor/pipelines/sequential/pipeline.py

@@ -72,8 +72,6 @@ def __call__(
                 calib_desc = f"({subgraph_index + 1}/{num_subgraphs}): Calibrating"
                 prop_desc = f"({subgraph_index + 1}/{num_subgraphs}): Propagating"
 
-                LifecycleCallbacks.sequential_epoch_start()
-
                 # do a preliminary pass to trigger modifier hooks
                 for batch_idx in tqdm.tqdm(range(len(dataloader)), desc=calib_desc):
                     inputs = intermediates.fetch(batch_idx, subgraph.input_names)