Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

metal : support permuted matrix multiplicaions #10033

Merged
merged 5 commits into from
Oct 25, 2024

Conversation

ggerganov
Copy link
Owner

@ggerganov ggerganov commented Oct 24, 2024

Generalize matrix multiplication kernels to support permuted tensors by taking into account the nb[1,2,3] strides.

TODO:

  • needs more testing
  • remove int step usage

@ggerganov ggerganov merged commit 6687503 into master Oct 25, 2024
54 checks passed
@ggerganov ggerganov deleted the gg/metal-mm-permute-support branch October 25, 2024 19:26
arthw pushed a commit to arthw/llama.cpp that referenced this pull request Nov 15, 2024
* metal : support permuted matrix multiplicaions

ggml-ci

* cont : use nb01 directly for row steps

ggml-ci

* cont : add comments [no ci]

* metal : minor refactor

* metal : minor
arthw pushed a commit to arthw/llama.cpp that referenced this pull request Nov 18, 2024
* metal : support permuted matrix multiplicaions

ggml-ci

* cont : use nb01 directly for row steps

ggml-ci

* cont : add comments [no ci]

* metal : minor refactor

* metal : minor
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant