search

liuliu / ccv

C-based/Cached/Core Computer Vision Library, A Modern Computer Vision Library
http://libccv.org
Other
7.07k stars 1.71k forks source link

MPS GEMM Backward #257

Closed weiyanlin117 closed 1 year ago

weiyanlin117 commented 1 year ago

implement GEMM backward with original MPS

h = g @ w.T dw = a.T @ g bias = g reduce on target axis.

10 tests original from cublas.tests.c all passed. Screenshot 2023-08-08 at 9 46 59 PM

weiyanlin117 commented 1 year ago

Screenshot 2023-08-10 at 12 49 06 AM