add new log line

Signed-off-by: Diego-Castan <[email protected]>

Author: diegocastanibm

vllm/v1/worker/gpu_model_runner.py

@@ -2558,6 +2558,8 @@ def freeze_gc():
                     desc="Capturing CUDA graph shapes")
             for num_tokens in compilation_cases:
                 # We skip EPLB here since we don't want to record dummy metrics
+                logger.info("DIEGO: compilation for number of tokens %d",
+                    num_tokens)
                 for _ in range(
                         self.compilation_config.cudagraph_num_of_warmups):
                     self._dummy_run(num_tokens,