add output format to ndk build func (apache#2999)

neo-ai · Apr 11, 2019 · d948a1f · d948a1f
1 parent 46df05e
commit d948a1f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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"