diff --git a/mlir/lib/ExecutionEngine/CRunnerUtils.cpp b/mlir/lib/ExecutionEngine/CRunnerUtils.cpp
--- a/mlir/lib/ExecutionEngine/CRunnerUtils.cpp
+++ b/mlir/lib/ExecutionEngine/CRunnerUtils.cpp
@@ -15,10 +15,12 @@
 #include "mlir/ExecutionEngine/CRunnerUtils.h"
 
 #ifndef _WIN32
+#include <alloca.h>
 #include <sys/time.h>
+#else
+#include "malloc.h"
 #endif // _WIN32
 
-#include <alloca.h>
 #include <cinttypes>
 #include <cstdio>
 #include <string.h>