]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
metal : add POOL2D and fix IM2COL (#9943)
authorJun Hee Yoo <redacted>
Wed, 23 Oct 2024 10:33:45 +0000 (19:33 +0900)
committerGitHub <redacted>
Wed, 23 Oct 2024 10:33:45 +0000 (13:33 +0300)
commit4c9388fb96ac2415fbb1239b7ba8346616606e2e
tree5a0b59dc3daac404fd5eaabcd231443efcf9d9af
parent873279b1592e433c4d9eb5065091cc98473c7bee
metal : add POOL2D and fix IM2COL (#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>
ggml/src/ggml-metal.m
ggml/src/ggml-metal.metal
tests/test-backend-ops.cpp