Felix Wang

icon company Google icon location XLA scaling, collective optimization.

Results 1 issues of Felix Wang

Update the xla commit to experiment potential fix for OOM

28 comment

Try to fix the large number of zero-broadcast generated from big pad in spmd partitioner. for sample HLO, ``` HloModule module_0076.reactant_loop_after_spmd_partitioner_cp_pattern ENTRY main() -> f64[3056,123] { %constant.946 = f64[] constant(0)...