Repository navigation
Add support for BLAS Syrk #1358
Description
Hi.
Thank you guys for this awesome library.
In my application, I need to compute a gram matrix (basically, x.t().dot(x)).
Using gemm is wasteful, as the result is symmetric and the lower half is redundant, so using syrk in this case is twice as fast as gemm.
On a general note, it would be really useful if the library could also easily support calling any general BLAS / LAPACK function using appropriate idioms (like getting memory layout, strides, etc.)
Reacted by bluss, Lishen1 and Benjamin Kay
Activity
nilgoyette commentedon Feb 6, 2024
This has already been discussed in this issue.
You probably already know but
general_mat_mulis "hidden" in linalg module. However this only offersgemm, which is not what you're asking for.tzachar commentedon Feb 7, 2024
I am aware of general_mat_mul.
Did anyone ever end up writing an additional crate for BLAS over ndarray?
[-]Add support for BLAS SYMM[/-][+]Add support for LAPACK Syrk[/+][-]Add support for LAPACK Syrk[/-][+]Add support for BLAS Syrk[/+]nilgoyette commentedon Feb 7, 2024
Afaik, nobody did. And I don't think anyone will do it in the near future. ndarray hasn't evolved much in the last 4-5 years because of a lack of maintainers.
Pencilcaseman commentedon Mar 19, 2024
I'd be interested in improving the BLAS functionality. Having spent a lot of time messing around with multidimensional array libraries myself (sole developer of LibRapid), I've got a decent understanding of how best to implement things.
As an experiment, I wrote a BLAS wrapper for Rust, which wraps a system BLAS library in a C API, which is further wrapped in a Rust API. It supports any BLAS library that has a CBLAS interface.
Implementing something like this (or using this directly??) could be interesting. Is that something worth looking into?
On an unrelated note, I'd also be interested in getting OpenCL and CUDA support added to this library -- an obvious first step would be BLAS with CLBlast and cuBLAS, but JIT-compiled kernels could be interesting, too.
nilgoyette commentedon Mar 19, 2024
This is something that @bluss might be interested in.
ajz34 commentedon Jul 14, 2024
Hi devs,
I found this issue when I'm working on some code to wrap BLAS (https://github.com/ajz34/blas-array2) recently. This crate is mostly for our own project (may not even useful due to our hesitation on using
ndarray), but is probably something you may be interested in.This crate can be seen as a proof-of-existence for Rust to implement Fortran95-interface-like API (https://netlib.org/blas/blast-forum/chapter2.pdf). It's more rust-like than C-FFI-like.
blas-array2acts as a wrapper and can handle SYRK (and other BLAS2/3 subroutines) with matrix defined byndarray, but not a solution built as member function inndarray(by comparasion,.dotfunction in numpy can detect whether SYRK could be utilized).