mirrors / karpathy--llm.c

文件历史

77 提交 53 分支 0 Git 标签
feature/cublaslt
提交图

2 次代码提交

作者 SHA1 备注 提交日期
Andrej Karpathy 8386e5393c intra-warp reductions and inter-warp shared memory reductions for B,T,C parallelism of softmax kernel nice 2024-04-09 20:08:11 +00:00
Andrej Karpathy c29d70a9b3 add softmax forward, both a naive parallel over B,T version, and a more optimized parallel over B,T,C version using shared memory for reductions 2024-04-09 19:22:47 +00:00