Support torch.compile in segment_matmul forward
#974
| Job | Run time |
|---|---|
| 3m 59s | |
| 3m 59s |
torch.compile in segment_matmul forward
#974
| Job | Run time |
|---|---|
| 3m 59s | |
| 3m 59s |