AnswerDotAI / gpu.cpp

A lightweight library for portable low-level GPU computation using WebGPU.
https://gpucpp.answer.ai
Apache License 2.0
3.68k stars 175 forks source link

Fix softmax kernel with padding for llm.c #58

Closed junjihashimoto closed 1 week ago

junjihashimoto commented 2 weeks ago

Fix the parameter of softmax-kernel. I added Cp of the extra parameter to softmax-kernel for llm.c. But I did not update the parameter of run.cpp.