mirror of
https://github.com/karpathy/llm.c.git
synced 2026-07-27 20:25:09 -04:00
Merge pull request #434 from ngc92/profile-fix
initialize multi_gpu_config so profile doesn't crash
This commit is contained in:
commit
00c291c679
1 changed files with 1 additions and 0 deletions
|
|
@ -49,6 +49,7 @@ int main(int argc, char *argv[]) {
|
|||
|
||||
// override number of layers to 1 because all layers repeat the same kernels, only profile once
|
||||
model.config.num_layers = 1;
|
||||
set_zero_configs(&multi_gpu_config, 0, model.num_parameters);
|
||||
|
||||
// do a training step
|
||||
gpt2_forward(&model, x, y, B, T);
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue