From a3801f01efae3434d6e4cdbef3dd455fcc10404f Mon Sep 17 00:00:00 2001
From: ademeure <arundemeure@gmail.com>
Date: Tue, 21 May 2024 16:53:11 +0100
Subject: [PATCH] added algorithm header for std::sort on windows (not sure
 about compile time impact...)

---
 train_gpt2.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train_gpt2.cu b/train_gpt2.cu
index 899293f75..16f8a4216 100644
--- a/train_gpt2.cu
+++ b/train_gpt2.cu
@@ -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