clpy
clpy copied to clipboard
Introduce more BLAS functions from CLBlast
Current ClPy supports only sgemm, which is manually implemented.
Candidate external libs:
- clBLAS https://github.com/clMathLibraries/clBLAS
- CLBLAST https://github.com/CNugteren/CLBlast
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
ALL BLAS Interface should be supported by ClPy is same as CuPy, not only used by ClPy itself.