diff --git a/llmc/cudnn_att.cpp b/llmc/cudnn_att.cpp index 8d0ad53..0330abe 100644 --- a/llmc/cudnn_att.cpp +++ b/llmc/cudnn_att.cpp @@ -3,6 +3,7 @@ // TODO this currently duplicates some of the utilities from the main file #define NOMINMAX +#include #include "cudnn_att.h" #include