From a4fc087a3ce92a7ea9000225a2a3859c6bd5d294 Mon Sep 17 00:00:00 2001 From: zjjott Date: Thu, 9 May 2024 14:33:56 +0800 Subject: [PATCH] fix log --- .../gpu/model/gpu_collective_performance_model.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/xla/service/gpu/model/gpu_collective_performance_model.cc b/xla/service/gpu/model/gpu_collective_performance_model.cc index dc2367240d026..f415b0e34d206 100644 --- a/xla/service/gpu/model/gpu_collective_performance_model.cc +++ b/xla/service/gpu/model/gpu_collective_performance_model.cc @@ -393,12 +393,12 @@ GpuPerformanceWithCollectiveModel::ComputeAllgatherTime( absl::Duration communication_time = absl::Milliseconds( std::max(intra_nodes_numel_bytes / (actual_bandwidth_ratio*intra_node_bus_bandwidth * 1e6), inner_node_numel_bytes / (actual_bandwidth_ratio*inner_node_bus_bandwidth * 1e6))); - std::cout << instr.ToString() << " numel_bytes:" << numel_bytes + VLOG(5) << instr.ToString() << " numel_bytes:" << numel_bytes << " intra_nodes_numel_bytes: " << intra_nodes_numel_bytes << " inner_node_numel_bytes: " << inner_node_numel_bytes << " intra_node_bus_bandwidth: " << intra_node_bus_bandwidth - << "Gbps,inner_node_bus_bandwidth: " << inner_node_bus_bandwidth - << "Gbps communication_time: " << communication_time; + << "GBps,inner_node_bus_bandwidth: " << inner_node_bus_bandwidth + << "GBps communication_time: " << communication_time; total_time += communication_time; return total_time; } @@ -434,8 +434,8 @@ GpuPerformanceWithCollectiveModel::ComputeReducescatterTime( << " intra_nodes_numel_bytes: " << intra_nodes_numel_bytes << " inner_node_numel_bytes: " << inner_node_numel_bytes << " intra_node_bus_bandwidth: " << intra_node_bus_bandwidth - << "Gbps,inner_node_bus_bandwidth: " << inner_node_bus_bandwidth - << "Gbps communication_time: " << communication_time; + << "GBps,inner_node_bus_bandwidth: " << inner_node_bus_bandwidth + << "GBps communication_time: " << communication_time; total_time += communication_time; return total_time; }