]> git.djapps.eu Git - pkg/ggml/sources/whisper.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 11:53:55 +0000 (14:53 +0300)
commit3eea171caba18bc1603d76d3dd6a3a94ade92bbc
tree91264de26221881bb5bf228c809be1808db440fc
parent64a56ebf13476249270fe64e20c5bad8bdc62912
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]