From ce55fbf62b65261e0ff1e96087c9a8b22d887ddf Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 16 Sep 2020 14:57:41 +0800
Subject: [PATCH] ci(imperative): run imperative python test

---
 .github/workflows/ci-cpu.yml  | 2 +-
 .github/workflows/ci-cuda.yml | 2 +-
 ci/run_python_test.sh         | 6 +++---
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/ci-cpu.yml b/.github/workflows/ci-cpu.yml
index 08c03ae60..9410cea7f 100644
--- a/.github/workflows/ci-cpu.yml
+++ b/.github/workflows/ci-cpu.yml
@@ -20,6 +20,6 @@ jobs:
       - name: Build MegEngine
         run: ./ci/cmake.sh cpu
       - name: Python test
-        run: ./ci/run_python_test.sh
+        run: ./ci/run_python_test.sh cpu
       - name: C++ test
         run: ./ci/run_cpp_test.sh cpu
diff --git a/.github/workflows/ci-cuda.yml b/.github/workflows/ci-cuda.yml
index e643bcb32..5d740ad5f 100644
--- a/.github/workflows/ci-cuda.yml
+++ b/.github/workflows/ci-cuda.yml
@@ -26,6 +26,6 @@ jobs:
       - name: Build MegEngine
         run: ./ci/cmake.sh cuda
       - name: Python test
-        run: ./ci/run_python_test.sh
+        run: ./ci/run_python_test.sh cuda
       - name: C++ test
         run: ./ci/run_cpp_test.sh cuda
diff --git a/ci/run_python_test.sh b/ci/run_python_test.sh
index 580d86781..a8add1d58 100755
--- a/ci/run_python_test.sh
+++ b/ci/run_python_test.sh
@@ -7,11 +7,11 @@ BASEDIR=$(readlink -f "$(dirname "$0")"/..)
 
 function python_test() {
     pip3 install --upgrade pip
-    pushd "${BASEDIR}"/python_module >/dev/null
+    pushd "${BASEDIR}"/imperative/python >/dev/null
         pip3 install -e '.[ci]'
         export PYTHONPATH=.
-        ./test/run.sh
+        ./test/run.sh $1
     popd >/dev/null
 }
 
-python_test
+python_test $1
-- 
GitLab