Skip to content

Batched SVD using cuSolver #14175

@jjbouza

Description

@jjbouza

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.

Metadata

Metadata

Assignees

Labels

featureA request for a proper, new feature.todoNot as important as medium or high priority tasks, but we will work on these.triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions