From c89c673594874d8ef58fcf310742dddd80d60874 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 6 Mar 2021 21:52:11 +0800
Subject: [PATCH] feat(lite): add windows lite whl scripts, enable
 megenginelite pytest and enable cpp test in ci

GitOrigin-RevId: fa963616a6e5d4aa548ac0f0854438c84d402116
---
 scripts/whl/windows/windows_build_whl.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/whl/windows/windows_build_whl.sh b/scripts/whl/windows/windows_build_whl.sh
index d7fd0b86..4a958c5a 100755
--- a/scripts/whl/windows/windows_build_whl.sh
+++ b/scripts/whl/windows/windows_build_whl.sh
@@ -104,6 +104,7 @@ function copy_more_dll() {
     fi
 }
 
+
 function do_build() {
     for ver in ${ALL_PYTHON}
     do
@@ -171,6 +172,7 @@ function do_build() {
         ${PYTHON_DIR}/python3 setup.py bdist_wheel
         cp ${BUILD_DIR}/staging/dist/Meg*.whl ${WINDOWS_WHL_HOME}/
 
+
         echo ""
         echo "##############################################################################################"
         echo "windows whl package location: ${WINDOWS_WHL_HOME}"
@@ -195,5 +197,10 @@ function third_party_prepare() {
 
 ######################
 export ALREADY_CONFIG_PYTHON_VER="yes"
+if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then
+    export SDK_NAME="cu101"
+else
+    export SDK_NAME="cpu"
+fi
 third_party_prepare
 do_build
-- 
GitLab