Support torch.compile in segment_matmul forward
#1035
| Job | Run time |
|---|---|
| 3s | |
| 3s |
torch.compile in segment_matmul forward
#1035
| Job | Run time |
|---|---|
| 3s | |
| 3s |