[PyTorch] Adding TP overlap support for te.Linear
with parallel_mode="column"
#6492
Job | Run time |
---|---|
22s | |
2m 3s | |
24s | |
22s | |
19s | |
55s | |
4m 25s |
te.Linear
with parallel_mode="column"
#6492
Job | Run time |
---|---|
22s | |
2m 3s | |
24s | |
22s | |
19s | |
55s | |
4m 25s |