diff --git a/include/ck/tensor_operation/gpu/device/device_base.hpp b/include/ck/tensor_operation/gpu/device/device_base.hpp
index 908ada016d..e3092bf9f2 100644
--- a/include/ck/tensor_operation/gpu/device/device_base.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_base.hpp
@@ -5,6 +5,7 @@
 
 #include <string>
 #include <sstream>
+#include <memory>
 
 #include "ck/stream_config.hpp"