[PyTorch] Adding TP overlap support for te.Linear
with parallel_mode="column"
#6449
Job | Run time |
---|---|
27m 52s | |
48m 41s | |
25m 11s | |
1h 3m 16s | |
2h 45m 0s |
te.Linear
with parallel_mode="column"
#6449
Job | Run time |
---|---|
27m 52s | |
48m 41s | |
25m 11s | |
1h 3m 16s | |
2h 45m 0s |