llm.c icon indicating copy to clipboard operation
llm.c copied to clipboard

Fix sizing typo in `train_gpt2_fp32.cu`

Open gajanan-choudhary opened this issue 6 months ago • 2 comments

Fixes a simple (harmless) typo that uses NUM_ACTIVATION_TENSORS instead of NUM_BACKWARD_TENSORS in train_gpt2_fp32.cu.

gajanan-choudhary avatar Aug 25 '24 00:08 gajanan-choudhary