Skip to content

Commit

Permalink
added algorithm header for std::sort on windows (not sure about compi…
Browse files Browse the repository at this point in the history
…le time impact...)
  • Loading branch information
ademeure committed May 21, 2024
1 parent b5e75dd commit a3801f0
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions train_gpt2.cu
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ This reads & runs in fp32, B=4, T=64, LR=1e-4, val/sample never (200),
#include <stdarg.h>
#include <string>
#include <vector>
#include <algorithm>
#include <functional>
#include <unordered_map>
// GPU / CUDA related
Expand Down

0 comments on commit a3801f0

Please sign in to comment.