I think it may be related to splatting the dimensions, if I replace dims... with explicit m, n in the original definition, I can restore the performance again.
I think it may be related to splatting the dimensions, if I replace dims... with explicit m, n in the original definition, I can restore the performance again.