[XLA:GPU] Fix merge utility not saving network throughput.

PiperOrigin-RevId: 733262741
jreiffers · Mar 4, 2025 · e50047f · e50047f
1 parent 17e913f
commit e50047f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xla/tools/collective_perf_table_gen.cc b/xla/tools/collective_perf_table_gen.cc
@@ -493,6 +493,8 @@ DeviceHloInstructionProfiles CollectivePerfTableGen::Merge(
     profile_proto.set_flops(profiling_result.flops);
     profile_proto.set_clock_cycles(profiling_result.clock_cycles);
     profile_proto.set_fingerprint(profiling_result.fingerprint);
+    profile_proto.set_network_throughput_bytes_per_sec(
+        profiling_result.network_throughput);
 
     *result.mutable_entries()->at(device_descriptor).add_entries() =
         std::move(profile_proto);