]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/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>
Fri, 1 Nov 2024 08:19:05 +0000 (10:19 +0200)
commita3231b2f2e6b8ae04f61faacf5dc33f83c8126dc
tree2be61df0b0d055d5468db9a61005d87e031cfb30
parent13db492f83bc3cdad3b34d1e24f029da73f77545
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>
ggml/src/ggml-metal.m
ggml/src/ggml-metal.metal