pytorch
243e135e - Sparse CSR CUDA: Add block sparse support for torch.triangular_solve (#68709)
Commit
View On GitHub
Commit
3 years ago
Sparse CSR CUDA: Add block sparse support for torch.triangular_solve (#68709) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68709 This PR adds support for triangular solver with a block CSR matrix. cc nikitaved pearu cpuhrsch IvanYashchuk ngimel Test Plan: Imported from OSS Reviewed By: ngimel Differential Revision: D33066067 Pulled By: cpuhrsch fbshipit-source-id: 9eaf1839071e9526be8d8c6d47732b24200f3557
Author
IvanYashchuk IvanYashchuk
Parents
5f3f327a
  • aten/src/ATen
    • cuda
      • File
        CUDASparse.h
      • File
        CUDASparseBlas.cpp
      • File
        CUDASparseBlas.h
      • File
        CUDASparseDescriptors.h
    • native/sparse/cuda
      • File
        SparseBlasImpl.cpp
  • test
    • File
      test_sparse_csr.py
FAQ Terms Privacy Refunds Impressum