Update on "[dtensor] Add single_dim_strategy functions for reduction-…