[PyTorch] Adding TP overlap support for te.Linear with parallel_mode="column"
#5870
| Job | Run time |
|---|---|
| 58s | |
| 58s |
te.Linear with parallel_mode="column"
#5870
| Job | Run time |
|---|---|
| 58s | |
| 58s |