pytorch
60eb1e53 - Sparse CSR CPU: Add block sparse support for MKL path (#68710)

Commit
4 years ago
Sparse CSR CPU: Add block sparse support for MKL path (#68710) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68710 This PR adds support for block sparse (BSR) matrices for functions that use Inspector-Executor MKL Sparse API. At the moment of this PR it's: * torch.addmm * torch.addmv * torch.triangular_solve (once https://github.com/pytorch/pytorch/pull/62180 is merged) cc nikitaved pearu cpuhrsch IvanYashchuk Test Plan: Imported from OSS Reviewed By: ZolotukhinM Differential Revision: D33179486 Pulled By: cpuhrsch fbshipit-source-id: e1dec0dccdbfed8b280be16b8c11fc9e770d50ae
Author
Parents
Loading