]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
metal : add POOL2D and fix IM2COL (llama/9943)
authorJun Hee Yoo <redacted>
Wed, 23 Oct 2024 10:33:45 +0000 (19:33 +0900)
committerGeorgi Gerganov <redacted>
Wed, 23 Oct 2024 17:28:03 +0000 (20:28 +0300)
commitcbcd08fd875c9ad25be7806d18160e0039189074
tree4fe5f239c0af1a25814a7d87b418cd64976e0b93
parentb327dbf54e7ea710b2c020ee00680a4082211921
metal : add POOL2D and fix IM2COL (llama/9943)

* add pool_2d

Signed-off-by: Junhee Yoo <redacted>
* fix im2col and add unittest for N>=1024

Signed-off-by: Junhee Yoo <redacted>
* add tests for N % 1024 != 0

Signed-off-by: Junhee Yoo <redacted>
* remove trailing whitespaces

Signed-off-by: Junhee Yoo <redacted>
* apply suggestions

Signed-off-by: Junhee Yoo <redacted>
* apply more optimization

- original IM2COL kernel + _ext with MIN()

Signed-off-by: Junhee Yoo <redacted>
* apply review: change kernel name of pool_2d

Signed-off-by: Junhee Yoo <redacted>
* apply review

Signed-off-by: Junhee Yoo <redacted>
* fix more formatting and enhance readability

Signed-off-by: Junhee Yoo <redacted>
---------

Signed-off-by: Junhee Yoo <redacted>
src/ggml-metal.m
src/ggml-metal.metal
tests/test-backend-ops.cpp