Support torch.compile in segment_matmul forward
#977
| Job | Run time |
|---|---|
| 5m 5s | |
| 5m 5s |
torch.compile in segment_matmul forward
#977
| Job | Run time |
|---|---|
| 5m 5s | |
| 5m 5s |