(self, bucket, dp_group)
| 2342 | tensor.orig_dense_tensor.copy_(tensor.to_dense()) |
| 2343 | |
| 2344 | def sparse_allreduce_bucket(self, bucket, dp_group): |
| 2345 | sparse_list = [] |
| 2346 | for sparse in bucket: |
| 2347 | sparse_list.append(self.sparse_allreduce(sparse, dp_group)) |
| 2348 | return sparse_list |
| 2349 | |
| 2350 | def sparse_allreduce(self, sparse, dp_group): |
| 2351 | original_data_type = sparse.values.dtype |
no test coverage detected