fix

apache · Mar 24, 2021 · 1eb8653 · 1eb8653
1 parent 556a922
commit 1eb8653
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py
@@ -1936,7 +1936,7 @@ def verify(
             ):
                 target += " -libs=thrust"
             for kind in ["graph", "debug"]:
-                intrp = relay.create_executor(kind, mod=mod, dev=dev, target=target)
+                intrp = relay.create_executor(kind, mod=mod, device=dev, target=target)
                 op_res = intrp.evaluate()(data)
                 tvm.testing.assert_allclose(
                     op_res.asnumpy(), ref_res.asnumpy(), rtol=1e-3, atol=1e-5

diff --git a/tests/scripts/task_python_docs.sh b/tests/scripts/task_python_docs.sh
@@ -51,6 +51,10 @@ find . -type f -path "*.log" | xargs rm -f
 find . -type f -path "*.pyc" | xargs rm -f
 make cython3
 
+# cleanup stale states for rust
+rm rust/tvm-sys/src/c_runtime_api.rs
+rm -rf rust/target/doc
+
 cd docs
 PYTHONPATH=`pwd`/../python make html |& tee /tmp/$$.log.txt
 if grep -E "failed to execute|Segmentation fault" < /tmp/$$.log.txt; then