ci(imperative): run imperative python test

.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

.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

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