diff --git a/core/src/index/CMakeLists.txt b/core/src/index/CMakeLists.txt
index 9125f6ea2c..bedd35a4a2 100644
--- a/core/src/index/CMakeLists.txt
+++ b/core/src/index/CMakeLists.txt
@@ -86,6 +86,10 @@ include(DefineOptionsCore)
 include(BuildUtilsCore)
 include(ThirdPartyPackagesCore)
 
+if (CUSTOMIZATION)
+    add_definitions(-DCUSTOMIZATION)
+endif (CUSTOMIZATION)
+
 add_subdirectory(knowhere)
 
 if (BUILD_COVERAGE STREQUAL "ON")
diff --git a/core/unittest/CMakeLists.txt b/core/unittest/CMakeLists.txt
index ac4fae85bb..b87a9a05f7 100644
--- a/core/unittest/CMakeLists.txt
+++ b/core/unittest/CMakeLists.txt
@@ -16,6 +16,9 @@
 # specific language governing permissions and limitations
 # under the License.
 #-------------------------------------------------------------------------------
+if (CUSTOMIZATION)
+    add_definitions(-DCUSTOMIZATION)
+endif (CUSTOMIZATION)
 
 include_directories("${CUDA_TOOLKIT_ROOT_DIR}/include")