mirrors / karpathy--llm.c

项目文件夹

karpathy--llm.c
文件
feature/openmpi_flexible_path
T
文件历史
Andrej Karpathy e12d090483 resolve merge conflicts
2024-06-26 04:05:46 +00:00
..
adamw.cuh
Make sure each parameter was rounded with a unique seed
2024-06-15 12:02:39 +00:00
attention.cuh
consolidate memory by deleting the grad activation struct, move some of the last pieces of memory to the forward pass
2024-06-14 00:54:58 +00:00
cublas_common.h
made implicit includes of standard headers explicit
2024-06-04 13:22:16 +03:00
cuda_common.h
make windows happy
2024-06-25 21:28:52 +03:00
cuda_utils.cuh
moved loss calculation to backward
2024-06-25 21:28:52 +03:00
cudnn_att.cpp
move all kernels into a main_stream in preparation for future parallelism
2024-06-04 22:35:12 +03:00
cudnn_att.h
move all kernels into a main_stream in preparation for future parallelism
2024-06-04 22:35:12 +03:00
dataloader.h
Remove -1, point at current sample
2024-06-18 13:35:17 +02:00
encoder.cuh
Add a comment for why pass a unique seed per param
2024-06-18 17:01:03 +02:00
fused_classifier.cuh
fix loss accumulation
2024-06-25 21:28:52 +03:00
gelu.cuh
Fix guarding logic, refactor
2024-06-17 14:12:05 +02:00
global_norm.cuh
update norm computation: multi-process reduce on GPU
2024-06-25 21:28:52 +03:00
layernorm.cuh
Merge pull request #600 from gordicaleksa/add_faster_layernorm_kernel
2024-06-17 14:19:22 -07:00
logger.h
Add learning rate and grad norm logging
2024-06-07 17:34:42 +02:00
matmul.cuh
make windows happy
2024-06-25 21:28:52 +03:00
mfu.h
Add H100 PCIe
2024-06-14 09:23:55 +02:00
outlier_detector.h
oops i was only using window size 16 for testing let's make it 128 or something
2024-06-25 01:56:04 +00:00
rand.h
Minor refactor of dataloader
2024-06-17 21:53:18 +02:00
sampler.h
refactor part 2, moving stuff into common files so that we can later nicely separate out all the kernels as well
2024-06-03 18:10:33 +00:00
schedulers.h
small touchups and add the wsd schedule
2024-06-21 00:49:01 +00:00
tokenizer.h
made implicit includes of standard headers explicit
2024-06-04 13:22:16 +03:00
utils.h
check token validity
2024-06-26 02:01:24 +03:00
zero.cuh
Fix windows include issue
2024-06-24 23:18:38 +02:00