[PyTorch] Adding TP overlap support for te.Linear
with parallel_mode="column"
#6493
Job | Run time |
---|---|
20s | |
21s | |
22s | |
1m 58s | |
21s | |
1m 3s | |
4m 25s |
te.Linear
with parallel_mode="column"
#6493
Job | Run time |
---|---|
20s | |
21s | |
22s | |
1m 58s | |
21s | |
1m 3s | |
4m 25s |