Add log message

Tabrizian · Tabrizian · commit 75e101ecd61e · 2025-07-18T10:35:04.000-07:00
Signed-off-by: Iman Tabrizian &lt;10105175+tabrizian@users.noreply.github.com&gt;
diff --git a/cpp/tensorrt_llm/thop/allreduceOp.cpp b/cpp/tensorrt_llm/thop/allreduceOp.cpp
@@ -163,6 +163,8 @@ class AllreduceOp
     {
         size_t size = input.numel();
         size_t seq_len = input.size(0);
+        size_t bytes_per_element = input.element_size();
+        TLLM_LOG_DEBUG("All reduce message size is %zu", size * bytes_per_element);
 
         if (std::getenv("TLLM_USE_NCCL_UB") && mStrategy == AllReduceStrategyType::UB)
         {

Original file line number	Diff line number	Diff line change
`@@ -163,6 +163,8 @@ class AllreduceOp`
`163`	`163`	`{`
`164`	`164`	`size_t size = input.numel();`
`165`	`165`	`size_t seq_len = input.size(0);`
	`166`	`+ size_t bytes_per_element = input.element_size();`
	`167`	`+ TLLM_LOG_DEBUG("All reduce message size is %zu", size * bytes_per_element);`
`166`	`168`
`167`	`169`	`if (std::getenv("TLLM_USE_NCCL_UB") && mStrategy == AllReduceStrategyType::UB)`
`168`	`170`	`{`