Add log message

Tabrizian · Tabrizian · commit da346eeadede · 2025-06-24T01:46:29.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
@@ -160,6 +160,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
`@@ -160,6 +160,8 @@ class AllreduceOp`
`160`	`160`	`{`
`161`	`161`	`size_t size = input.numel();`
`162`	`162`	`size_t seq_len = input.size(0);`
	`163`	`+ size_t bytes_per_element = input.element_size();`
	`164`	`+ TLLM_LOG_DEBUG("All reduce message size is %zu", size * bytes_per_element);`
`163`	`165`
`164`	`166`	`if (std::getenv("TLLM_USE_NCCL_UB") && mStrategy == AllReduceStrategyType::UB)`
`165`	`167`	`{`