Fix the device issue with argpartition
in torch
#2539
Job | Run time |
---|---|
10m 51s | |
2m 0s | |
14m 39s | |
9m 34s | |
4m 39s | |
41m 43s |
argpartition
in torch
#2539
Job | Run time |
---|---|
10m 51s | |
2m 0s | |
14m 39s | |
9m 34s | |
4m 39s | |
41m 43s |