diff --git a/CHANGELOG.md b/CHANGELOG.md
index 327b1f9b3e426f73f5ffad5447c11d0519520ce1..590d5c151c5169fb9ad95e6c67cfd7bfbb57a413 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,7 @@ Please mark all change in change log and use the ticket from JIRA.
 ## Bug
 - \#246 - Exclude src/external folder from code coverage for jenkin ci
 - \#248 - Reside src/external in thirdparty
+- \#327 - Search does not use GPU when index type is FLAT
 
 ## Feature
 - \#12 - Pure CPU version for Milvus
diff --git a/core/src/scheduler/optimizer/OnlyGPUPass.cpp b/core/src/scheduler/optimizer/OnlyGPUPass.cpp
index 3fcda0e8a347bc6ca62890c78b394314f426c9e1..2a72f9757e9b10b4b08d09597d5454af2ca00cee 100644
--- a/core/src/scheduler/optimizer/OnlyGPUPass.cpp
+++ b/core/src/scheduler/optimizer/OnlyGPUPass.cpp
@@ -34,7 +34,8 @@ OnlyGPUPass::Run(const TaskPtr& task) {
 
     auto search_task = std::static_pointer_cast<XSearchTask>(task);
     if (search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFSQ8 &&
-        search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFFLAT) {
+        search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFFLAT &&
+        search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IDMAP) {
         return false;
     }