Possible to extend to 4x4 matrices?

[Jessie-jx]

Hi @banctilrobitaille , thank you for your open-source sharing and I found that this is really helpful to accelerate 3x3 batch eigendecomposition. Actually, I am facing a problem of calculating the eigenvalue and eigenvectors of a 61440x4x4 tensor, 61440 is batch size, 4x4 is matrix size. So I'm wondering whether this fast algorithm can be extend to 4x4 matrices? If so, could you provide me some detailed suggestions?