Skip to content

np.dot(a, a.T) should be detected and dispatched to the BLAS syrk routine #6794

@njsmith

Description

@njsmith

BLAS provides a specialized routine for the common A @ A.T operation. We should detect this case (same data pointer, reversed strides) and dispatch to syrk instead of gemm. (see)

In addition to providing a nice speedup, this should presumably avoid issues like #6793, where dot(A, A.T) returned a non-exactly-symmetric matrix and this caused confusion.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions