]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
feat: cuda implementation for `ggml_conv_transpose_1d` (#854)
authorJohn Balis <redacted>
Tue, 2 Jul 2024 16:09:52 +0000 (11:09 -0500)
committerGitHub <redacted>
Tue, 2 Jul 2024 16:09:52 +0000 (18:09 +0200)
commit84376bded7508797cde269b09de7e2deebfb8208
tree696f84094a10baff982a46018a1d3657d7feb42d
parent052f2e93a94906d597eaf697a4c72ece33f3504b
feat: cuda implementation for `ggml_conv_transpose_1d` (#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>
src/ggml-cuda.cu
src/ggml-cuda/conv-transpose-1d.cu [new file with mode: 0644]
src/ggml-cuda/conv-transpose-1d.cuh [new file with mode: 0644]
tests/CMakeLists.txt
tests/test-backend-ops.cpp
tests/test-conv-transpose-1d.cpp [new file with mode: 0644]