[Feature]: Add NN layout for gradlib #1533
Description
Suggestion Description
Hi guys, now only F.linear(mk, nk -> TN layout) can be replaced by tgemm.mm. What about torch.matmul and torch.addmm？(mk,kn -> NN layout)。 Will this be supported in the future?
Operating System
No response
GPU
No response
ROCm Component
No response