Add masked_index_select and refactor masked_index_put #5142
| Job | Run time |
|---|---|
| 6s | |
| 20m 6s | |
| 1h 14m 4s | |
| 1h 54m 1s | |
| 1h 47m 43s | |
| 1h 39m 34s | |
| 1h 45m 15s | |
| 34s | |
| 41s | |
| 44s | |
| 45s | |
| 28s | |
| 31s | |
| 8h 44m 32s |
| Job | Run time |
|---|---|
| 6s | |
| 20m 6s | |
| 1h 14m 4s | |
| 1h 54m 1s | |
| 1h 47m 43s | |
| 1h 39m 34s | |
| 1h 45m 15s | |
| 34s | |
| 41s | |
| 44s | |
| 45s | |
| 28s | |
| 31s | |
| 8h 44m 32s |