diff --git a/python/tvm/contrib/ndk.py b/python/tvm/contrib/ndk.py index 931fa03a7308..e1703ce03f8e 100644 --- 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"