From d048c11aa04aa0b1b687bb1581c51dab8787c21d Mon Sep 17 00:00:00 2001
From: qiuzhongya <qiuzhongya@huawei.com>
Date: Sat, 29 Oct 2022 09:58:54 +0800
Subject: [PATCH] support for incredibuild

---
 build.bat                      | 33 ++++++++++++++++++++++++++-------
 mindspore/ccsrc/CMakeLists.txt |  3 +++
 2 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/build.bat b/build.bat
index 3c6a745e8b9..da9af092e05 100644
--- a/build.bat
+++ b/build.bat
@@ -23,7 +23,9 @@ SET BUILD_PATH=%BASE_PATH%/build
 
 SET threads=8
 SET ENABLE_GITEE=OFF
-
+SET ENABLE_INCREDIBUILD=OFF
+SET ENABLE_MSVC=OFF
+set BUILD_TYPE=Release
 set VERSION_STR=''
 for /f "tokens=1" %%a in (version.txt) do (set VERSION_STR=%%a)
 
@@ -37,6 +39,22 @@ IF "%FROM_GITEE%" == "1" (
     SET ENABLE_GITEE=ON
 )
 
+ECHO %1%|FINDSTR "^ms_vs"
+IF %errorlevel% == 0 (
+    echo "use msvc compiler"
+    SET ENABLE_MSVC=ON
+) else (
+    echo "use mingw compiler"
+)
+
+where buildconsole
+IF %errorlevel% == 0 (
+    echo "use buildconsole to speed up compile"
+    SET ENABLE_INCREDIBUILD=ON
+) else (
+    echo "fail to find buildconsole"
+)
+
 IF NOT EXIST "%BUILD_PATH%" (
     md "build"
 )
@@ -70,6 +88,7 @@ IF "%1%" == "lite" (
         echo "======Start gen VS2019 Project for MS cpu debug======"
         cmake -DCMAKE_BUILD_TYPE=Debug -DDEBUG_MODE=ON -DENABLE_CPU=ON -DENABLE_MINDDATA=ON -DUSE_GLOG=ON -DENABLE_GITEE=%ENABLE_GITEE% ^
             -G "Visual Studio 16 2019" -A x64 ../..
+        set BUILD_TYPE=Debug
     ) ELSE (
         echo "======Start gen MinGW64 Project for MS cpu ======"
         cmake -DCMAKE_BUILD_TYPE=Release -DENABLE_CPU=ON -DENABLE_MINDDATA=ON -DUSE_GLOG=ON -DENABLE_GITEE=%ENABLE_GITEE% ^
@@ -82,12 +101,12 @@ IF NOT %errorlevel% == 0 (
     EXIT /b 1
 )
 
-IF "%1%" == "ms_vs_gpu" (
-    cmake --build . --config Release --target package
-) ELSE IF "%1%" == "ms_vs_cpu" (
-    cmake --build . --config Release --target package
-) ELSE IF "%1%" == "ms_vs_cpu_debug" (
-    cmake --build . --config Debug --target package
+IF ON == %ENABLE_MSVC% (
+    IF ON == %ENABLE_INCREDIBUILD% (
+        buildconsole /command="cmake --build . --config %BUILD_TYPE% --target package"
+    ) ELSE (
+        cmake --build . --config %BUILD_TYPE% --target package
+    )
 ) ELSE (
     cmake --build . --target package -- -j%threads%
 )
diff --git a/mindspore/ccsrc/CMakeLists.txt b/mindspore/ccsrc/CMakeLists.txt
index 244bd98704d..c58ff7b02d2 100644
--- a/mindspore/ccsrc/CMakeLists.txt
+++ b/mindspore/ccsrc/CMakeLists.txt
@@ -94,6 +94,9 @@ function(set_nvcc_flag CUDA_NVCC_FLAGS)
         list(APPEND CUDA_NVCC_FLAGS -arch=sm_53)
     endif()
     list(APPEND CUDA_NVCC_FLAGS --expt-relaxed-constexpr)
+    if(MSVC)
+        list(APPEND CUDA_NVCC_FLAGS -t0)
+    endif()
     set(${CUDA_NVCC_FLAGS} ${${CUDA_NVCC_FLAGS}} PARENT_SCOPE)
 endfunction()