add output format to ndk build func (#2999)

diff --git a/python/tvm/contrib/ndk.py b/python/tvm/contrib/ndk.py
index 931fa03..e1703ce 100644 (file)
--- a/python/tvm/contrib/ndk.py
+++ b/python/tvm/contrib/ndk.py
@@ -63,3 +63,6 @@ def create_shared(output,
         msg = "Compilation error:\n"
         msg += py_str(out)
         raise RuntimeError(msg)
+
+# assign output format
+create_shared.output_format = "so"