(item)
| 240 | return int(start) |
| 241 | |
| 242 | def get_partition_size(item): |
| 243 | global mp_rank, mp_size, mp_group |
| 244 | size = item.numel() |
| 245 | partition_size = size/mp_size |
| 246 | return int(partition_size) |
| 247 | |
| 248 | def get_full_inputs(tensors): |
| 249 | inputs=[] |
no outgoing calls
no test coverage detected