fix msvc build

InternLM · Oct 31, 2023 · fbdcc1e · fbdcc1e
1 parent 3bcc7e3
commit fbdcc1e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/turbomind/models/llama/Barrier.h b/src/turbomind/models/llama/Barrier.h
@@ -2,6 +2,7 @@
 
 #pragma once
 
+#include "src/turbomind/utils/cuda_utils.h"
 #include "src/turbomind/utils/logger.h"
 #ifndef _MSC_VER
 #include <pthread.h>