-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
feat: adds ggml_pad_ext
to allow prefix padding
#864
base: master
Are you sure you want to change the base?
Conversation
I forgot that |
…d prefix and postfix padding with cuda backend
Alright so it supports up to 4d prefix and postfix padding with the CPU version. for cuda it supports up to 3d prefix and postfix padding (the cuda pad kernel was previously limited to 3d tensors and I didn't remove this limitation). Two of the tests in This modification leaves the arguments of Ready for review :^) |
ggml_pad
to allow prefix paddingggml_pad_ext
to allow prefix padding
ggml-ci
(I noticed I need to change some variable names in |
this and pad_reflect_1d are ready for review again :^) |
modifies
ggml_pad
so you can specify beginning and end padding for all 4 dimensions of a tensor.Previously, padding could only be added at the end of the first 2 dimensions.The padding is still constant padding with a value of zero, as before.Adds new test file
test-pad.cpp