Add a field for pcie bandwidth in SchedulerConfig

PiperOrigin-RevId: 728835685
openxla · Feb 25, 2025 · 13e2c53 · 13e2c53
1 parent 2efc815
commit 13e2c53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xla/service/latency_hiding_scheduler.h b/xla/service/latency_hiding_scheduler.h
@@ -150,6 +150,7 @@ struct SchedulerConfig {
   int64_t max_hops_to_closest_selective_overlap = 0;
   int64_t rerun = 0;
   int64_t parallel_collective_overlap_limit = 1;
+  double pcie_bandwidth_in_bytes_per_second = 6.0 * 1e9;
 };
 
 // Class used estimate latency between instructions and cost of HLOs.