Skip to content

Reimplement GEMM kernels #6

Description

@softcookiepp

Right now, the GEMM function for this library relies on kernels ported from CLBlast.
While these have reasonable-enough performance, they do not take advantage of dedicated matmul acceleration instruction sets present on some modern GPUs, provided by VK_KHR_cooperative_matrix.
Implementing a GEMM kernel that uses this capability would result in significant performance improvements on applicable hardware.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions