Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix "dim_zero_cat" reduction Lightning-AI#2225
Fixes Lightning-AI#2225 by changing "reduced = global_state + local_state" to "reduced = torch.cat([global_state, local_state]
- Loading branch information