]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
feat: cuda implementation for `ggml_conv_transpose_1d` (ggml/854)
authorJohn Balis <redacted>
Tue, 2 Jul 2024 16:09:52 +0000 (11:09 -0500)
committerGeorgi Gerganov <redacted>
Mon, 8 Jul 2024 09:23:00 +0000 (12:23 +0300)
commitfde13b3bb9f569f07fd8af74696ee48a43d05131
tree9cb214f47d31271b6e1946787947aec30756a3ba
parent470939d483d1c89b7292f78bac1fd27c42c171ce
feat: cuda implementation for `ggml_conv_transpose_1d` (ggml/854)

* conv transpose 1d passing test for 1d input and kernel

* working for different input and output channel counts, added test for variable stride

* initial draft appears to work with stride other than 1

* working with all old and new conv1d  tests

* added a test for large tensors

* removed use cuda hardcoding

* restored test-conv-transpose.c

* removed unused arugments, and fixed bug where test failure would cause subsequent tests to fail

* fixed accumulator bug

* added test to test-backend-ops

* fixed mistake

* addressed review

* fixed includes

* removed blank lines

* style and warning fixes

* return failure when test fails

* fix supports_op

---------

Co-authored-by: slaren <redacted>
ggml/src/ggml-cuda.cu
ggml/src/ggml-cuda/conv-transpose-1d.cu [new file with mode: 0644]
ggml/src/ggml-cuda/conv-transpose-1d.cuh [new file with mode: 0644]
tests/test-backend-ops.cpp