clpy icon indicating copy to clipboard operation
clpy copied to clipboard

Introduce more BLAS functions from CLBlast

Open LWisteria opened this issue 6 years ago • 3 comments

Current ClPy supports only sgemm, which is manually implemented.

LWisteria avatar Jul 11 '18 09:07 LWisteria

Candidate external libs:

  • clBLAS https://github.com/clMathLibraries/clBLAS
  • CLBLAST https://github.com/CNugteren/CLBlast

LWisteria avatar Aug 22 '18 07:08 LWisteria

Latest CuPy uses

  • {s,d}geam (core/core.pyx) - doesn't exists in clBlast
  • {c,z}gemm (core/core.pyx)
  • {s,d,c,z}trsm (linalg/solve.py)
  • gemmEx (core/core.pyx)
  • {s,d,c,z}gemmStridedBatched (core/core.pyx) - exists in clBlast

nsakabe-fixstars avatar Oct 25 '18 02:10 nsakabe-fixstars

ALL BLAS Interface should be supported by ClPy is same as CuPy, not only used by ClPy itself.

LWisteria avatar Nov 15 '18 07:11 LWisteria