-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Description
It seems there are several people working on batch mode linear algebra routines, i.e. #11796 and #14071 are active.
Any plans for adding a batch mode SVD? This would be useful for certain implementations of group equivariant networks.
I'm not completely familiar with the PyTorch codebase, but if I'm not mistaken the usual backend used for linear algebra computations on the GPU is MAGMA. I don't think MAGMA implements a batch SVD operation, but cuSolver does for small matrices (max 32x32). For larger matrices we can just fall back to the current approach.
If no one else is planning on working on this I can take a look at it. The correct way to do this would be to model something like #9949, right?
I realize several others have made similar suggestions: #10172, #4689. Those issues don't seem active however.