patch log function name (#168)

Co-authored-by: Kyle Sayers <[email protected]>
Co-authored-by: Dipika Sikka <[email protected]>

Author: 3 people

src/llmcompressor/modifiers/quantization/gptq/utils/gptq_wrapper.py

@@ -314,23 +314,24 @@ def _log_metrics(self, start_tick: float, losses: torch.Tensor):
         :param start_tick: time when algorithm started"
         :param losses: loss as result of algorithm
         """
-        logger.log("METRIC", "time %.2f" % (time.time() - start_tick))
-        logger.log("METRIC", "error %.2f" % torch.sum(losses).item())
+        patch = logger.patch(lambda r: r.update(function="compress"))
+        patch.log("METRIC", "time %.2f" % (time.time() - start_tick))
+        patch.log("METRIC", "error %.2f" % torch.sum(losses).item())
 
         gpu_usage = get_GPU_memory_usage()
         if len(gpu_usage) > 0:
             for i in range(len(gpu_usage)):
                 perc = gpu_usage[i][0] * 100
                 total_memory = int(gpu_usage[i][1])  # GB
-                logger.log(
+                patch.log(
                     "METRIC",
                     (
                         f"GPU {i} | usage: {perc:.2f}%"
                         f" | total memory: {total_memory} GB"
                     ),
                 )
 
-        logger.log(
+        patch.log(
             "METRIC",
             f"Compressed layer size: {get_layer_size_bytes(self.layer)} MB",
         )