[PyTorch] Adding TP overlap support for te.Linear with parallel_mode="column"
#4389
| Job | Run time |
|---|---|
| 6s | |
| 3m 11s | |
| 6s | |
| 3m 23s |
te.Linear with parallel_mode="column"
#4389
| Job | Run time |
|---|---|
| 6s | |
| 3m 11s | |
| 6s | |
| 3m 23s |