]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
ggml: new gpu kernels + extends ggml_leaky_relu + ggml_pad (#621)
authorSteward Garcia <redacted>
Wed, 13 Dec 2023 14:08:48 +0000 (09:08 -0500)
committerGitHub <redacted>
Wed, 13 Dec 2023 14:08:48 +0000 (09:08 -0500)
commit5bf85a52211a2e4a17a257eed8a24ac6ea75627a
tree77717e574939bb88109244d49e4ccf982cd3f7ca
parenta027a92c1db83ec3cd157cb4e36c6d88cf0615bc
ggml: new gpu kernels + extends ggml_leaky_relu + ggml_pad (#621)

* add new cuda kernels and new op ggml_pad

* add ggml_tanh cuda kernel

* remove old broadcast impl

* restore some changes

* cuda: optimized im2col + group_norm kernels

* extent ggml_leaky -> ggml_leaky_relu

* fix some code issues

* cuda: concat support 4 dims

* cuda: fix ggml_acc + add backends ops test

* restore ggml_pad + add backend op test

* metal : implement GGML_OP_ACC

* ggml : fix bug in ggml_upscale

* metal : add ggml_upscale

* metal : add ggml_tanh

* metal : add ggml_gelu_quick

* ggml : make ggml_pad more general purpose

* metal : add ggml_pad

* ggml_leaky_relu as regular op + fix identation

* cuda: ggml_acc admit all op_parms

* negative_slope better pass param

* metal : add ggml_leaky_relu

* metal : add ggml_group_norm

* cuda : minor

* ggml : add GGML_OP_LEAKY_RELU to ggml_compute_backward

* metal : soft max, tanh, supports_op fixes

* test-backend-ops : add sentinels between tensors to detect overflows

---------

Co-authored-by: Georgi Gerganov <redacted>
Co-authored-by: slaren <redacted>
.gitignore
examples/yolo/yolov3-tiny.cpp
include/ggml/ggml.h
src/CMakeLists.txt
src/ggml-cuda.cu
src/ggml-metal.m
src/ggml-metal.metal
src/ggml.c
tests/test-backend-ops.cpp
tests/test-conv1d.cpp
tests/test-conv2d.cpp