fix lint issue and comments.

xuhancn · xuhancn · commit 3a8736ebba48 · 2024-10-20T18:14:49.000+08:00
diff --git a/aten/src/ATen/mkl/MklAllocationHelp.cpp b/aten/src/ATen/mkl/MklAllocationHelp.cpp
@@ -2,12 +2,17 @@
 #include <ATen/Config.h>
 
 #if AT_MKLDNN_ENABLED()
+#ifdef USE_MIMALLOC_ON_MKL
 #include <c10/core/impl/alloc_cpu.h>
 
+/*
+MKL have a method to register memory allocation APIs via i_malloc.h, High
+performance memory allocation APIs will help improve MKL performance.
+Please check MKL online document：
+https://www.intel.com/content/www/us/en/docs/onemkl/developer-guide-windows/2024-2/redefining-memory-functions.html
+*/
 #include <i_malloc.h>
 
-#ifdef USE_MIMALLOC_ON_MKL
-
 bool register_mimalloc_api_to_mkl()
 {
     i_malloc  = c10::mi_malloc_wrapper::c10_mi_malloc;
@@ -21,4 +26,3 @@ bool register_mimalloc_api_to_mkl()
 static bool g_b_registered_mkl_alloction = register_mimalloc_api_to_mkl();
 #endif
 #endif
-
diff --git a/cmake/Summary.cmake b/cmake/Summary.cmake
@@ -164,7 +164,7 @@ function(caffe2_print_configuration_summary)
   message(STATUS "  USE_OPENMP            : ${USE_OPENMP}")
   message(STATUS "  USE_MIMALLOC          : ${USE_MIMALLOC}")
   if(${USE_MIMALLOC})
-    message(STATUS "    USE_MIMALLOC_ON_MKL          : ${USE_MIMALLOC_ON_MKL}")
+    message(STATUS "    USE_MIMALLOC_ON_MKL   : ${USE_MIMALLOC_ON_MKL}")
   endif()
   message(STATUS "  USE_VULKAN            : ${USE_VULKAN}")
   if(${USE_VULKAN})
